Spaces:

Lianglan
/

NLLB200-distilled-1.3B

Running

Lianglan commited on Feb 27, 2023

Commit

bbd346a

•

1 Parent(s): a1e0123

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,8 @@ from langs import LANGS
 #from langs_all import LANGS  ##for 200+ languages
 TASK = "translation"
-CKPT = "facebook/nllb-200-distilled-600M"
 model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
 tokenizer = AutoTokenizer.from_pretrained(CKPT)

 #from langs_all import LANGS  ##for 200+ languages
 TASK = "translation"
+CKPT = "facebook/nllb-200-distilled-1.3B"
+#CKPT = "facebook/nllb-200-distilled-600M"
 model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
 tokenizer = AutoTokenizer.from_pretrained(CKPT)