update model for ndebele-english translation

app.py

@@ -1,52 +1,35 @@
 import gradio as gr
-from gradio.mix import Parallel, Series
-
 from transformers import pipeline
 
-translater = pipeline("translation", model="VietAI/envit5-translation")
-
+translater_nr_en = pipeline("translation", model="dsfsi/nr-en-m2m100-gov", src_lang="nr", tgt_lang="en")
 
-def translate(inp, direction):
-    if direction == 'en->vi':
-        text = "en: " + inp
-    else:
-        text = "vi: " + inp
-        
-    res = translater(
-        text,
-        max_length=512,
-        early_stopping=True,
-    )[0]['translation_text'][3:]
+def translate(inp):
+    # Translate from isiNdebele to English
+    res = translater_nr_en(inp, max_length=512, early_stopping=True)[0]['translation_text']
     return res
 
 description = """
 <p>
 <center>
-Multi-domain Translation Between English and Vietnamese
+One-way Translation from isiNdebele to 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:heraclex12@gmail.com' target='_blank'>Hieu Tran</a></p></center></p>"
+article = "<p style='text-align: center'><a href='https://huggingface.co/dsfsi/nr-en-m2m100-gov' 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"]
+    ["Ngiyabonga kakhulu ngesipho osinike sona."],
+    ["Ukuthula kuhlale kuyindlela ephilayo yempilo yethu."]
 ]
+
 iface = gr.Interface(
     fn=translate,
-
-    title="🌸MTet Translation🌸",
+    title="isiNdebele to English 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")
+    inputs=gr.components.Textbox(lines=5, placeholder="Enter isiNdebele text (maximum 5 lines)", label="Input"),
+    outputs="text"
+)
 
-iface.launch(enable_queue=True)
+iface.launch(enable_queue=True)