Spaces:

hci-lab-dcug
/

UG_Maternal_Health_Translate

Running

nyarkssss commited on 27 days ago

Commit

2d6cdff

1 Parent(s): 681bb6d

add

Files changed (1) hide show

nllb.py CHANGED Viewed

@@ -10,8 +10,8 @@ model_dict = {}
 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"
     }[model_name]
     print("\tLoading model: %s" % model_name)

 def load_models(model_name: str):
     # build model and tokenizer
     model_name_dict = {
+        "ug_entw_translate": "nyarkssss/MMT_ENGLISH_AKAN_05",
+        "ug_twen_translate": "nyarkssss/MMT_AKAN_ENGLISH_05"
     }[model_name]
     print("\tLoading model: %s" % model_name)