Spaces:

UNESCO
/

nllb

Running on Zero

davanstrien HF staff commited on Apr 4

Commit

6bcde50

•

1 Parent(s): 5cb6981

Update MODEL_NAME in app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import platform
 import torch
 device = "cpu" if platform.system() == "Darwin" else "cuda"
-MODEL_NAME = "facebook/m2m100_1.2B"
 code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[1]))
 flores_codes = list(code_mapping.keys())

 import torch
 device = "cpu" if platform.system() == "Darwin" else "cuda"
+MODEL_NAME = "facebook/nllb-200-3.3B"
 code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[1]))
 flores_codes = list(code_mapping.keys())