Spaces:

ashourzadeh7
/

nllb-translation-demo-2

Paused

ashourzadeh7 commited on Jul 7, 2024

Commit

c090998

verified ·

1 Parent(s): 02bf3e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import torch
 # this model was loaded from https://hf.co/models
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
-model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M").to(device)
-tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 LANGS = ["pes_Arab", "ckb_Arab", "eng_Latn"]

 # this model was loaded from https://hf.co/models
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-3.3B").to(device)
+tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-3.3B")
 LANGS = ["pes_Arab", "ckb_Arab", "eng_Latn"]