Update app.py

app.py

@@ -1,16 +1,19 @@
-from transformers import AutoModelForSeq2SeqLM,AutoTokenizer
+from transformers import AutoModelForSeq2SeqLM,AutoTokenizer, pipeline
 import gradio as grad
 mdl_name = "oddadmix/masrawy-english-arabic-translator"
-mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
-my_tkn = AutoTokenizer.from_pretrained(mdl_name)
+# mdl = AutoModelForSeq2SeqLM.from_pretrained(mdl_name)
+# my_tkn = AutoTokenizer.from_pretrained(mdl_name)
 
-#opus_translator = pipeline("translation", model=mdl_name)
+pipe = pipeline("translation", model=mdl_name)
 
 def translate(text):
-    inputs = my_tkn(text, return_tensors="pt")
-    trans_output = mdl.generate(**inputs)
-    response = my_tkn.decode(trans_output[0], skip_special_tokens=True)
+    # inputs = my_tkn(text, return_tensors="pt")
+    # trans_output = mdl.generate(**inputs)
+    response = pipe(text)[0]['translation_text']
 
     #response = opus_translator(text)
     return response
-grad.Interface(translate, inputs=["text",], outputs="text").launch()
+
+input_textbox = grad.Textbox(lines=5, placeholder="Enter text to translate", label="Input Text")
+output_textbox = grad.Textbox(lines=5, label="Egyption Arabic Translation")
+grad.Interface(translate, inputs=[input_textbox], outputs=output_textbox).launch()