Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -40,6 +40,7 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
|
|
| 40 |
return generated_text
|
| 41 |
|
| 42 |
def predict_t5(text, model, tokenizer, temperature=1.2):
|
|
|
|
| 43 |
input_ids = tokenizer.encode(text, return_tensors="pt")
|
| 44 |
|
| 45 |
model.eval()
|
|
|
|
| 40 |
return generated_text
|
| 41 |
|
| 42 |
def predict_t5(text, model, tokenizer, temperature=1.2):
|
| 43 |
+
text = text.replace('\n','')
|
| 44 |
input_ids = tokenizer.encode(text, return_tensors="pt")
|
| 45 |
|
| 46 |
model.eval()
|