Spaces:
Runtime error
Runtime error
ilhamsyahids
commited on
Commit
•
2f4418e
1
Parent(s):
0ea36d0
use nllb-distilled-1.3B model
Browse filesSigned-off-by: Ilham Syahid S <ilhamsyahids@gmail.com>
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,
|