Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import MarianMTModel, MarianTokenizer
 st.title("🌐 Universal Language Translator App")
 st.write("""
 Translate text from any language to any other language using an open-source multilingual model.  
-This app supports many languages and ensures a user-friendly experience.
+This app supports a wide range of languages with ease of use.
 """)
 
 # Instructions
@@ -58,15 +58,13 @@ if st.button("Translate"):
         st.error("Please enter some text to translate.")
     else:
         try:
-            # Define model name dynamically
-            model_name = f"Helsinki-NLP/opus-mt-{languages[source_language]}-{languages[target_language]}"
-            
-            # Load tokenizer and model
+            # Universal Model for Language Translation
+            model_name = f"Helsinki-NLP/opus-mt-mul-mul"
             tokenizer = MarianTokenizer.from_pretrained(model_name)
             model = MarianMTModel.from_pretrained(model_name)
             
-            # Translate
-            inputs = tokenizer(source_text, return_tensors="pt", padding=True, truncation=True)
+            # Translation
+            inputs = tokenizer(f">>{languages[target_language]}<< {source_text}", return_tensors="pt", padding=True, truncation=True)
             translated_tokens = model.generate(**inputs)
             translated_text = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
             
@@ -74,7 +72,7 @@ if st.button("Translate"):
             st.subheader("🔄 Translated Text:")
             st.text_area("Translation Output:", value=translated_text, height=150)
         except Exception as e:
-            st.error("Translation failed. Please ensure the selected languages are supported.")
+            st.error("Translation failed. Ensure the text and languages are valid.")
             st.error(str(e))
 
 # Footer