update to use siswati-english models

app.py

@@ -1,40 +1,33 @@
 import gradio as gr
-from gradio.mix import Parallel, Series
-
 from transformers import pipeline
 
-translater = pipeline("translation", model="VietAI/envit5-translation")
-
+translater_en_ss = pipeline("translation", model="dsfsi/en-ss-m2m100-combo")
+translater_ss_en = pipeline("translation", model="dsfsi/ss-en-m2m100-combo")
 
 def translate(inp, direction):
-    if direction == 'en->vi':
-        text = "en: " + inp
+    if direction == 'en->ss':
+        res = translater_en_ss(inp, max_length=512, early_stopping=True)[0]['translation_text']
     else:
-        text = "vi: " + inp
-        
-    res = translater(
-        text,
-        max_length=512,
-        early_stopping=True,
-    )[0]['translation_text'][3:]
+        res = translater_ss_en(inp, max_length=512, early_stopping=True)[0]['translation_text']
     return res
 
 description = """
 <p>
 <center>
-Multi-domain Translation Between English and Vietnamese
+Multi-domain Translation Between Siswati and English
 </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>"
+article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/en-ss-m2m100-combo' target='_blank'>by dsfsi</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"]
+    ["Thank you for your help", "en->ss"],
+    ["Ngiyabonga ngesiciniseko sakho", "ss->en"]
 ]
+
 iface = gr.Interface(
     fn=translate,
-
-    title="🌸MTet Translation🌸",
+    title="Siswati-English Translation",
     description=description,
     article=article,
     examples=examples,
@@ -42,11 +35,12 @@ iface = gr.Interface(
         gr.inputs.Textbox(lines=5, placeholder="Enter text (maximum 5 lines)", label="Input"),
         gr.inputs.Radio(
             choices=[
-                'en->vi',
-                'vi->en'],
-            default='en->vi',
+                'en->ss',
+                'ss->en'],
+            default='en->ss',
             label='Direction'),
     ],
-    outputs="text")
+    outputs="text"
+)
 
-iface.launch(enable_queue=True)
+iface.launch(enable_queue=True)