k1ngtai commited on
Commit
ca36bdf
·
verified ·
1 Parent(s): 45f8a2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,9 +8,10 @@ from flores200_codes import flores_codes
8
 
9
  def load_models():
10
  # build model and tokenizer
11
- model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
 
12
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
13
- #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
14
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
15
  }
16
 
 
8
 
9
  def load_models():
10
  # build model and tokenizer
11
+ model_name_dict = {
12
+ #'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
13
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
14
+ 'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
15
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
16
  }
17