keithhon commited on
Commit
68d06c3
1 Parent(s): cf7dafa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,8 +11,8 @@ def load_models():
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
 
18
  model_dict = {}
@@ -29,7 +29,7 @@ def load_models():
29
 
30
  def translation(source, target, text):
31
  if len(model_dict) == 2:
32
- model_name = 'nllb-distilled-3.3B'
33
 
34
  start_time = time.time()
35
  source = flores_codes[source]
 
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
 
18
  model_dict = {}
 
29
 
30
  def translation(source, target, text):
31
  if len(model_dict) == 2:
32
+ model_name = 'nllb-distilled-1.3B'
33
 
34
  start_time = time.time()
35
  source = flores_codes[source]