Spaces:
Running
Running
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()
|