davanstrien HF staff commited on
Commit
1473813
1 Parent(s): 126fd42

Update MODEL_NAME and description in app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
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 = "facebook/nllb-200-distilled-600M"
 
 
 
 
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: