gokaygokay
/

Flux-Prompt-Enhance

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

gokaygokay commited on Aug 26

Commit

5309247

•

1 Parent(s): 4b8c3b6

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -8,7 +8,8 @@ language:
 base_model: google-t5/t5-base
 pipeline_tag: text2text-generation
 ---
-```
 from transformers import pipeline
 enhancer = pipeline('text2text-generation', model=model, tokenizer=tokenizer, repetition_penalty=1.5, device="cuda")

 base_model: google-t5/t5-base
 pipeline_tag: text2text-generation
 ---
+```python
 from transformers import pipeline
 enhancer = pipeline('text2text-generation', model=model, tokenizer=tokenizer, repetition_penalty=1.5, device="cuda")