Update translation.py

translation.py

@@ -37,7 +37,7 @@ def translate(text, source_lang, target_lang):
     if src_code == "en" and tgt_code == "fr":
         tokenizer, model = DEFAULT_TOKENIZER, DEFAULT_MODEL
     else:
-        tokenizer, model = load_model(src_code, tgt_lang)
+        tokenizer, model = load_model(src_code, tgt_code) 
     
     # Perform translation
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=400)