Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
1473813
1
Parent(s):
126fd42
Update MODEL_NAME and description in app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,11 @@ from flores import code_mapping
|
|
5 |
import platform
|
6 |
|
7 |
device = "cpu" if platform.system() == "Darwin" else "cuda"
|
8 |
-
MODEL_NAME =
|
|
|
|
|
|
|
|
|
9 |
|
10 |
code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[1]))
|
11 |
flores_codes = list(code_mapping.keys())
|
@@ -55,7 +59,8 @@ def translate(text: str, src_lang: str, tgt_lang: str):
|
|
55 |
|
56 |
|
57 |
description = """
|
58 |
-
No Language Left Behind (NLLB) is a series of open-source models aiming to provide high-quality translations between 200 language.
|
|
|
59 |
|
60 |
|
61 |
with gr.Blocks() as demo:
|
|
|
5 |
import platform
|
6 |
|
7 |
device = "cpu" if platform.system() == "Darwin" else "cuda"
|
8 |
+
MODEL_NAME = (
|
9 |
+
"facebook/nllb-200-distilled-600M"
|
10 |
+
if platform.system() == "Darwin"
|
11 |
+
else "facebook/nllb-200-3.3B"
|
12 |
+
)
|
13 |
|
14 |
code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[1]))
|
15 |
flores_codes = list(code_mapping.keys())
|
|
|
59 |
|
60 |
|
61 |
description = """
|
62 |
+
No Language Left Behind (NLLB) is a series of open-source models aiming to provide high-quality translations between 200 language.
|
63 |
+
"""
|
64 |
|
65 |
|
66 |
with gr.Blocks() as demo:
|