0x7o commited on
Commit
d360479
1 Parent(s): 2d930f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -9,7 +9,7 @@ from flores200_codes import flores_codes
9
 
10
  def load_models():
11
  # build model and tokenizer
12
- model_name_dict = {'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B'}
13
 
14
  model_dict = {}
15
 
@@ -65,9 +65,9 @@ if __name__ == '__main__':
65
 
66
  outputs = gr.JSON()
67
 
68
- title = "NLLB distilled 1.3B demo"
69
 
70
- demo_status = "Demo is running on A10G GPU"
71
  description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
72
  examples = [
73
  ['English', 'Russian', 'Hi. nice to meet you']
9
 
10
  def load_models():
11
  # build model and tokenizer
12
+ model_name_dict = {'nllb-distilled-1.3B': 'facebook/nllb-200-3.3B'}
13
 
14
  model_dict = {}
15
 
65
 
66
  outputs = gr.JSON()
67
 
68
+ title = "NLLB 3.3B demo"
69
 
70
+ demo_status = "Demo is running on GPU"
71
  description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
72
  examples = [
73
  ['English', 'Russian', 'Hi. nice to meet you']