Update app.py

app.py

@@ -59,14 +59,21 @@ LANGUAGES = {
 
 # Function to get the appropriate translation model and tokenizer
 def get_translation_model(source_lang, target_lang):
+    # Use a generic model for translation if specific model is not available
     model_name = f"Helsinki-NLP/opus-mt-{source_lang}-{target_lang}"
-    model = MarianMTModel.from_pretrained(model_name)
-    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    try:
+        model = MarianMTModel.from_pretrained(model_name)
+        tokenizer = MarianTokenizer.from_pretrained(model_name)
+    except OSError:
+        st.error(f"Model '{model_name}' not found. Please check the model name or use another language pair.")
+        return None, None
     return model, tokenizer
 
 # Function to translate text
 def translate_text(text, source_lang, target_lang):
     model, tokenizer = get_translation_model(source_lang, target_lang)
+    if model is None or tokenizer is None:
+        return ""
     inputs = tokenizer([text], return_tensors="pt", truncation=True)
     translated_ids = model.generate(inputs['input_ids'], max_length=1024)
     translated_text = tokenizer.decode(translated_ids[0], skip_special_tokens=True)