Update src/translation.py

src/translation.py

@@ -88,9 +88,9 @@ def translate_facebook(s:str, src_iso:str, dest_iso:str)-> str:
     set_seed(555) 
 
     # Load model
-    tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M",src_lang=f"{src_iso}_Latn") #use_auth_token=True, 
+    tokenizer = AutoTokenizer.from_pretrained("anyantudre/NLLB-finetuned-fr-to-mos-V3",src_lang=f"{src_iso}_Latn") #use_auth_token=True, 
                                               
-    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M") #, use_auth_token=True)
+    model = AutoModelForSeq2SeqLM.from_pretrained("anyantudre/NLLB-finetuned-fr-to-mos-V3") #, use_auth_token=True)
 
     # Inference
     encoded = tokenizer(s, return_tensors="pt")