yvankob commited on
Commit
448682c
·
1 Parent(s): e2819e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ pipe = pipeline(
29
  def load_models():
30
  # build model and tokenizer
31
  model_name_dict = {
32
- 'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
33
- #'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
34
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
35
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
36
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
 
29
  def load_models():
30
  # build model and tokenizer
31
  model_name_dict = {
32
+ 'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-600M',
33
+ #'nllb-distilled-600M': 'facebook/nllb-200-distilled-1.3B',
34
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
35
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
36
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',