Update app.py

app.py

@@ -4,12 +4,12 @@ import docx2txt
 from transformers import pipeline
 import sentencepiece
 
-# Initialize translation pipelines
+# Load translation models
 def load_translation_models():
-    """Load translation models with error handling."""
+    """Load translation models for English and Urdu."""
     try:
         translator_en = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en", framework="pt")  
-        translator_ur = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-ur", framework="pt")
+        translator_ur = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ur", framework="pt")
         return translator_en, translator_ur
     except Exception as e:
         st.error(f"Error initializing translation models: {e}")