Update app.py

app.py

@@ -12,8 +12,11 @@ def translate(text):
     translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return translated_text
 
+def predict(text):
+    return translate(text)
+
 iface = gr.Interface(
-    fn=translate,
+    fn=predict,
     inputs=gr.Textbox(lines=5, label="输入文本 (支持中英互译)"),
     outputs=gr.Textbox(label="翻译结果"),
     title="WiNGPT-Babel 翻译 Demo",