Create app.py

app.py

@@ -0,0 +1,18 @@
+import gradio as gr
+import os
+
+os.system("pip install transformers sentencepiece torch")
+
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("alirezamsh/small100")
+model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
+
+def fn(text, lang):
+    tokenizer.tgt_lang = lang
+    encoded = tokenizer(text, return_tensors="pt")
+    generated_tokens = model.generate(**encoded)
+    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+
+demo = gr.Interface(fn=fn, inputs=["text", "text"], outputs="text")
+demo.launch()