Update app.py

app.py

@@ -1,19 +1,27 @@
 import spaces
 from transformers import AutoModelForSeq2SeqLM,AutoTokenizer, pipeline
 import gradio as grad
-mdl_name = "oddadmix/bi-msa-egyptian-translator-masrawy"
+mdl_name = "oddadmix/Masrawy-BiLingual-v1"
 pipe = pipeline("translation", model=mdl_name, device = 'cuda')
 
 @spaces.GPU
-def translate(text):
+def translate(text, direction):
     # inputs = my_tkn(text, return_tensors="pt")
     # trans_output = mdl.generate(**inputs)
-    
-    response = pipe(text)[0]['translation_text']
+    if direction == "MSA → Egyptian":
+        prompt = text + " <msa_to_egy>"
+    else:
+        prompt = text + " <egy_to_msa>"    
+    response = pipe(prompt)[0]['translation_text']
 
     #response = opus_translator(text)
     return response
 
-input_textbox = grad.Textbox(lines=5, placeholder="Enter Modern Standard Arabic text to translate", label="Input Text")
-output_textbox = grad.Textbox(lines=5, label="Egyptian Arabic Translation")
-grad.Interface(translate, inputs=[input_textbox], outputs=output_textbox).launch()
+input_textbox = grad.Textbox(lines=5, placeholder="اكتب النص هنا بالعربية الفصحى أو باللهجة المصرية...", label="Input Text")
+output_textbox = grad.Textbox(lines=5, label="النص المترجم")
+grad.Interface(translate, inputs=[input_textbox,         grad.Radio(
+            choices=["MSA → Egyptian", "Egyptian → MSA"], 
+            value="MSA → Egyptian", 
+            label="اتجاه الترجمة"
+        )], outputs=output_textbox, title="Masrawy: MSA ↔ Egyptian Translator",
+    description="اختر اتجاه الترجمة وأدخل النص بالعربية الفصحى أو باللهجة المصرية.").launch()