galamallikarjun commited on
Commit
a5a124e
·
verified ·
1 Parent(s): 6bb50ea

Upload 6 files

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