Update app.py

app.py

@@ -5,9 +5,13 @@ from transformers import pipeline
 
 translater = pipeline("translation", model="VietAI/envit5-translation")
 
- 
-def translate(inp):
-    text = "en: " + inp
+
+def translate(inp, direction):
+    if direction == 'en->vi':
+        text = "en: " + inp
+    else:
+        text = "vi: " + inp
+        
     res = translater(
         text,
         max_length=512,
@@ -15,23 +19,34 @@ def translate(inp):
     )[0]['translation_text'][3:]
     return res
 
-
-
-sample_url = [['VietAI is a non-profit organization with the mission of nurturing AI talents and building a community of world-class AI experts in Vietnam.'],]
-
-article = "<p style='text-align: center'><a href='https://vietai.org' target='_blank'>by VietAI Research</a> | <a href='https://github.com/vietai/mtet' target='_blank'>Github</a> | Contact: <a href='mailto:[email protected]' target='_blank'>Hieu Tran</a></p></center></p>"
-
-
-iface = gr.Interface(fn=translate, 
-  inputs = gr.inputs.Textbox(
-      lines = 5,
-      label = 'Enter an article...'
-  ),
-  outputs = 'text',
-  title = 'En<->Vi MTet Translation',
-  theme = 'grass',
-  layout = 'horizontal',
-  article=article,
-  examples=sample_url)
-
-iface.launch()
+description = """
+<p>
+<center>
+Multi-domain Translation Between English and Vietnamese
+</center>
+</p>
+"""
+article = "<p style='text-align: center'><a href='http://translate.vietai.org' target='_blank'>by VietAI Research</a> | <a href='https://github.com/vietai/mTet' target='_blank'>Github</a> | Contact: <a href='mailto:[email protected]' target='_blank'>Hieu Tran</a></p></center></p>"
+examples = [
+    ["Dear God, thank you for granting us the evergreen garden of this world", "en->vi"],
+    ["Thuốc này đã bị cấm sử dụng trong ngành thú y tại Ấn Độ.", "vi->en"]
+]
+iface = gr.Interface(
+    fn=translate,
+
+    title="🌸MTet Translation🌸",
+    description=description,
+    article=article,
+    examples=examples,
+    inputs=[
+        gr.inputs.Textbox(lines=5, placeholder="Enter text (maximum 5 lines)", label="Input"),
+        gr.inputs.Radio(
+            choices=[
+                'en->vi',
+                'vi->en'],
+            default='en->vi',
+            label='Direction'),
+    ],
+    outputs="text")
+
+iface.launch(enable_queue=True)