ilhamsyahids commited on
Commit
2f4418e
1 Parent(s): 0ea36d0

use nllb-distilled-1.3B model

Browse files

Signed-off-by: Ilham Syahid S <ilhamsyahids@gmail.com>

Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -7,9 +7,9 @@ from flores200_codes import flores_codes
7
  def load_models():
8
  # build model and tokenizer
9
  model_name_dict = {
10
- "nllb-distilled-600M": "facebook/nllb-200-distilled-600M",
11
  "nllb-distilled-1.3B": "facebook/nllb-200-distilled-1.3B",
12
- "nllb-1.3B": "facebook/nllb-200-1.3B",
13
  # "nllb-3.3B": "facebook/nllb-200-3.3B",
14
  }
15
 
@@ -68,9 +68,9 @@ if __name__ == "__main__":
68
  inputs = [
69
  gr.inputs.Radio(
70
  [
71
- "nllb-distilled-600M",
72
  "nllb-distilled-1.3B",
73
- "nllb-1.3B",
74
  # "nllb-3.3B"
75
  ],
76
  label="NLLB Model",
@@ -87,7 +87,7 @@ if __name__ == "__main__":
87
 
88
  demo_status = "Demo is running on CPU"
89
  description = f"Using NLLB model, details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
90
- examples = [["nllb-1.3B", "Najdi Arabic", "English", "جلست اطفال"]]
91
 
92
  gr.Interface(
93
  translation,
 
7
  def load_models():
8
  # build model and tokenizer
9
  model_name_dict = {
10
+ # "nllb-distilled-600M": "facebook/nllb-200-distilled-600M",
11
  "nllb-distilled-1.3B": "facebook/nllb-200-distilled-1.3B",
12
+ # "nllb-1.3B": "facebook/nllb-200-1.3B",
13
  # "nllb-3.3B": "facebook/nllb-200-3.3B",
14
  }
15
 
 
68
  inputs = [
69
  gr.inputs.Radio(
70
  [
71
+ # "nllb-distilled-600M",
72
  "nllb-distilled-1.3B",
73
+ # "nllb-1.3B",
74
  # "nllb-3.3B"
75
  ],
76
  label="NLLB Model",
 
87
 
88
  demo_status = "Demo is running on CPU"
89
  description = f"Using NLLB model, details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
90
+ examples = [["nllb-distilled-1.3B", "Najdi Arabic", "English", "جلست اطفال"]]
91
 
92
  gr.Interface(
93
  translation,