new

app.py

@@ -8,7 +8,7 @@ nllb_translate = gr.Interface(
     fn=translation,
     inputs=[
         gr.Dropdown(
-            ["ug_twen_translate", "ug_entw_translate", "nllb-200-distilled-600M"],
+            ["ug_twen_translate", "ug_entw_translate"],
             label="Model",
             value="ug_twen_translate",
         ),
@@ -26,8 +26,8 @@ nllb_translate = gr.Interface(
     ],
     outputs="json",
     examples=NLLB_EXAMPLES,
-    title="NLLB Translation Demo",
-    description="Translate text from one language to another.",
+    title="UG MatCare Translation Demo",
+    description="Translate text from one language to another, focused on maternal health care.",
     allow_flagging="never",
 )
 

nllb.py

@@ -11,8 +11,7 @@ def load_models(model_name: str):
     # build model and tokenizer
     model_name_dict = {
         "ug_entw_translate": "nyarkssss/ug_entw_translate",
-        "ug_twen_translate": "nyarkssss/ug_twen_translate",
-        "nllb-200-distilled-600M": "facebook/nllb-200-distilled-600M",
+        "ug_twen_translate": "nyarkssss/ug_twen_translate"
     }[model_name]
 
     print("\tLoading model: %s" % model_name)