Spaces:
Running
Running
anyantudre
commited on
Commit
•
3943aca
1
Parent(s):
ebf5733
Update src/translation.py
Browse files- src/translation.py +1 -1
src/translation.py
CHANGED
@@ -88,7 +88,7 @@ def translate_facebook(s:str, src_iso:str, dest_iso:str)-> str:
|
|
88 |
set_seed(555)
|
89 |
|
90 |
# Load model
|
91 |
-
tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",src_lang=f"{src_iso}_Latn"
|
92 |
|
93 |
model = AutoModelForSeq2SeqLM.from_pretrained("anyantudre/NLLB-finetuned-fr-to-mos-V3") #, use_auth_token=True)
|
94 |
|
|
|
88 |
set_seed(555)
|
89 |
|
90 |
# Load model
|
91 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",src_lang=f"{src_iso}_Latn") #use_auth_token=True,
|
92 |
|
93 |
model = AutoModelForSeq2SeqLM.from_pretrained("anyantudre/NLLB-finetuned-fr-to-mos-V3") #, use_auth_token=True)
|
94 |
|