Svngoku commited on
Commit
f4f48e3
1 Parent(s): 8480bd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,8 +9,8 @@ from flores200_codes import flores_codes
9
  def load_models():
10
  # build model and tokenizer
11
  model_name_dict = {
12
- 'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
13
- #'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
14
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
15
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
16
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
@@ -70,7 +70,7 @@ if __name__ == '__main__':
70
 
71
  outputs = gr.outputs.JSON()
72
 
73
- title = "NLLB distilled 1.3B demo"
74
 
75
  demo_status = "Demo is running on CPU"
76
  description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
 
9
  def load_models():
10
  # build model and tokenizer
11
  model_name_dict = {
12
+ #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
13
+ 'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
14
  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
15
  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
16
  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
 
70
 
71
  outputs = gr.outputs.JSON()
72
 
73
+ title = "NLLB distilled 600M demo endpoint"
74
 
75
  demo_status = "Demo is running on CPU"
76
  description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"