khof312 commited on
Commit
944775a
1 Parent(s): 2d1fac5

Remove use of authentication token.

Browse files
Files changed (1) hide show
  1. src/translation.py +3 -2
src/translation.py CHANGED
@@ -87,8 +87,9 @@ def translate_facebook(s:str, src_iso:str, dest_iso:str)-> str:
87
  set_seed(555)
88
 
89
  # Load model
90
- tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M", use_auth_token=True, src_lang=f"{src_iso}_Latn")
91
- model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M", use_auth_token=True)
 
92
 
93
  # Inference
94
  encoded = tokenizer(s, return_tensors="pt")
 
87
  set_seed(555)
88
 
89
  # Load model
90
+ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",src_lang=f"{src_iso}_Latn") #use_auth_token=True,
91
+
92
+ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M") #, use_auth_token=True)
93
 
94
  # Inference
95
  encoded = tokenizer(s, return_tensors="pt")