Update translation.py

translation.py

@@ -13,7 +13,7 @@ def _load_default_model():
 # Cache other models dynamically
 @st.cache_resource
 def load_model(src_lang, tgt_lang):
-    """Load MarianMT model for a specific language pair."""
+    """Load the MarianMT model and tokenizer for a language pair."""
     model_name = f"Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}"
     try:
         tokenizer = MarianTokenizer.from_pretrained(model_name)
@@ -37,16 +37,16 @@ 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_code)
+        tokenizer, model = load_model(src_code, tgt_lang)
     
     # Perform translation
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True, max_length=400)
     translated = model.generate(**inputs)
     return tokenizer.decode(translated[0], skip_special_tokens=True)
 
-# Language dictionary (limited for speed)
+# Dictionary of supported languages with MarianMT codes
 LANGUAGES = {
-   "English": "en",
+    "English": "en",
     "French": "fr",
     "Spanish": "es",
     "German": "de",