mipatov commited on
Commit
53f4560
1 Parent(s): 90ca31e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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()