anyantudre commited on
Commit
3943aca
1 Parent(s): ebf5733

Update src/translation.py

Browse files
Files changed (1) hide show
  1. 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", use_fast=False) #use_auth_token=True,
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