Update app.py

app.py

@@ -23,7 +23,6 @@ languages = {
     "Dutch": "nl", "Korean": "ko", "Swedish": "sv", "Turkish": "tr", "Polish": "pl", "Danish": "da",
     "Finnish": "fi", "Norwegian": "no", "Greek": "el", "Hebrew": "he", "Thai": "th", "Vietnamese": "vi",
     "Czech": "cs", "Hungarian": "hu", "Indonesian": "id", "Malay": "ms", "Filipino": "tl", "Swahili": "sw",
-    "Urdu": "ur"
 }
 
 # Translation Functionality
@@ -33,8 +32,15 @@ if operation == "Translate Text":
 
     if st.button("Translate"):
         try:
-            translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-{target}".format(
-                target=languages[target_language]))
+            if source_language == "English":
+                model_name = "Helsinki-NLP/opus-mt-en-mul"
+            elif target_language == "English":
+                model_name = "Helsinki-NLP/opus-mt-mul-en"
+            else:
+                st.error("Direct translation between non-English languages is not supported yet.")
+                st.stop()
+
+            translator = pipeline("translation", model=model_name)
             translation = translator(text, max_length=400)[0]['translation_text']
             st.write("### Translation:")
             st.write(translation)