ilhamsyahids commited on
Commit
c50283a
1 Parent(s): afb6c6a

comment 3.3B

Browse files

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

Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -10,7 +10,7 @@ def load_models():
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
 
16
  model_dict = {}
@@ -67,7 +67,12 @@ if __name__ == "__main__":
67
  lang_codes = list(flores_codes.keys())
68
  inputs = [
69
  gr.inputs.Radio(
70
- ["nllb-distilled-600M", "nllb-distilled-1.3B", "nllb-1.3B", "nllb-3.3B"],
 
 
 
 
 
71
  label="NLLB Model",
72
  default="nllb-distilled-600M",
73
  ),
@@ -82,7 +87,7 @@ if __name__ == "__main__":
82
 
83
  demo_status = "Demo is running on CPU"
84
  description = f"Using NLLB model, details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
85
- examples = [["nllb-3.3B", "Najdi Arabic", "English", "جلست اطفال"]]
86
 
87
  gr.Interface(
88
  translation,
 
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
 
16
  model_dict = {}
 
67
  lang_codes = list(flores_codes.keys())
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",
77
  default="nllb-distilled-600M",
78
  ),
 
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,