YeMinNaing commited on
Commit
d582114
·
verified ·
1 Parent(s): 7f7ed1e

device cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from functools import lru_cache
12
 
13
  nltk.download("punkt_tab")
14
  REMOVED_TARGET_LANGUAGES = {"Ligurian", "Lombard", "Sicilian"}
15
- device = "cpu" if platform.system() == "Darwin" else "cuda"
16
  MODEL_NAME = "facebook/nllb-200-3.3B"
17
  code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[0]))
18
  flores_codes = list(code_mapping.keys())
 
12
 
13
  nltk.download("punkt_tab")
14
  REMOVED_TARGET_LANGUAGES = {"Ligurian", "Lombard", "Sicilian"}
15
+ device = "cpu"
16
  MODEL_NAME = "facebook/nllb-200-3.3B"
17
  code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[0]))
18
  flores_codes = list(code_mapping.keys())