Ashish Ranjan Karn commited on
Commit
7d3036e
·
1 Parent(s): 74202e5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
21
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
22
  tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
23
  device = -1 #0 if torch.cuda.is_available() else -1
24
- LANGS = ["ace_Arab", "eng_Latn", "fra_Latn", "spa_Latn", "hin_Deva"]
25
  langs_mapper = {
26
  "English": "eng_Latn",
27
  "French": "fra_Latn",
 
21
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
22
  tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
23
  device = -1 #0 if torch.cuda.is_available() else -1
24
+ LANGS = ["eng_Latn", "fra_Latn", "spa_Latn", "hin_Deva"]
25
  langs_mapper = {
26
  "English": "eng_Latn",
27
  "French": "fra_Latn",