Spaces:
Sleeping
Sleeping
Ashish Ranjan Karn
commited on
Commit
·
7d3036e
1
Parent(s):
74202e5
fix
Browse files
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 = ["
|
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",
|