Update app.py

app.py

@@ -148,10 +148,10 @@ import gradio as grad
 text2text_tkn= T5Tokenizer.from_pretrained("t5-small")
 mdl = T5ForConditionalGeneration.from_pretrained("t5-small")
 def text2text_translation(text):
-     inp = "translate English to Portuguese:: "+text
-     enc = text2text_tkn(inp, return_tensors="pt")
-     tokens = mdl.generate(**enc)
-     response=text2text_tkn.batch_decode(tokens)
+     inp = "translate English to Portuguese: "+text
+     enc = text2text_tkn(inp, return_tensors="pt", max_length=512, truncation=True)
+     tokens = mdl.generate(**enc, max_length=100, num_return_sequences=1, early_stopping=True)
+     response=text2text_tkn.decode(tokens[0], skip_special_tokens=True)
      return response
 para=grad.Textbox(lines=1, label="English Text", placeholder="Text in English")
 out=grad.Textbox(lines=1, label="Portuguese Translation")