htukor commited on
Commit
e732240
1 Parent(s): ba372e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from langs import LANGS
6
 
7
  TASK = "translation"
8
  CKPT = "facebook/nllb-200-distilled-600M"
9
-
10
  model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
11
  tokenizer = AutoTokenizer.from_pretrained(CKPT)
12
 
 
6
 
7
  TASK = "translation"
8
  CKPT = "facebook/nllb-200-distilled-600M"
9
+ CKPT = "facebook/nllb-200-distilled-1.3B"
10
  model = AutoModelForSeq2SeqLM.from_pretrained(CKPT)
11
  tokenizer = AutoTokenizer.from_pretrained(CKPT)
12