Update app.py

app.py

@@ -8,11 +8,11 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("alirezamsh/small100")
 model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
 
-def fn(text):
-    tokenizer.tgt_lang = "en"
+def fn(text, lang):
+    tokenizer.tgt_lang = lang
     encoded_zh = tokenizer(text, return_tensors="pt")
     generated_tokens = model.generate(**encoded_zh)
     return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 
-demo = gr.Interface(fn=fn, inputs="text", outputs="text")
+demo = gr.Interface(fn=fn, inputs="text", outputs=["text", "text"])
 demo.launch()