nyarkssss commited on
Commit
48bc5bc
·
1 Parent(s): 7ac78a1

UI updates

Browse files
Files changed (2) hide show
  1. flores200_codes.py +1 -1
  2. nllb.py +1 -1
flores200_codes.py CHANGED
@@ -2,5 +2,5 @@
2
  flores_codes = {
3
  "Akan": "aka_Latn",
4
  "English": "eng_Latn",
5
- "Twi": "twi_Latn"
6
  }
 
2
  flores_codes = {
3
  "Akan": "aka_Latn",
4
  "English": "eng_Latn",
5
+ # "Twi": "twi_Latn"
6
  }
nllb.py CHANGED
@@ -10,7 +10,7 @@ model_dict = {}
10
  def load_models(model_name: str):
11
  # build model and tokenizer
12
  model_name_dict = {
13
- "model_2": "nyarkssss/model_2",
14
  "ug_twen_translate": "nyarkssss/ug_twen_translate",
15
  "nllb-200-distilled-600M": "facebook/nllb-200-distilled-600M",
16
  }[model_name]
 
10
  def load_models(model_name: str):
11
  # build model and tokenizer
12
  model_name_dict = {
13
+ "ug_entw_translate": "nyarkssss/ug_entw_translate",
14
  "ug_twen_translate": "nyarkssss/ug_twen_translate",
15
  "nllb-200-distilled-600M": "facebook/nllb-200-distilled-600M",
16
  }[model_name]