Update app.py

app.py

@@ -5,9 +5,9 @@ mdl = MT5ForConditionalGeneration.from_pretrained("google/mt5-small")
 tokenizer = AutoTokenizer.from_pretrained("google/mt5-small")
 
 def translation_CN2EN(text):
-    inp = "translate English to German: " + text
+    inp =  "translate Chinese to English: " + text
     enc = tokenizer(inp, return_tensors="pt").input_ids
-    tokens = mdl.generate(enc)
+    tokens = mdl(**enc)
     response = tokenizer.decode(tokens[0], skip_special_tokens=True)
     return response
 para=grad.Textbox(lines=1, label="Chinese Text", placeholder="Text in Chinese")