ahmadsipra73 commited on
Commit
89ca97d
·
1 Parent(s): 0c2ae6b

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import MT5ForConditionalGeneration, MT5Tokenizer
3
+
4
+ model_name = "SeyedAli/English-To-Persian-Translation-mT5-V1"
5
+ tokenizer = MT5Tokenizer.from_pretrained(model_name)
6
+ model = MT5ForConditionalGeneration.from_pretrained(model_name)
7
+
8
+ text_input = gr.TextArea(label="جمله انگلیسی",text_align="right",rtl=True,type="text")
9
+ text_output = gr.TextArea(label="ترجمه فارسی",text_align="right",rtl=True,type="text")
10
+
11
+ def Translate(text, **generator_args):
12
+ input_ids = tokenizer.encode(text, return_tensors="pt")
13
+ res = model.generate(input_ids, **generator_args)
14
+ output = tokenizer.batch_decode(res, skip_special_tokens=True)[0]
15
+ return output
16
+
17
+ iface = gr.Interface(fn=Translate, inputs=text_input, outputs=text_output)
18
+ iface.launch(share=False)