Update app.py

app.py

@@ -1,14 +1,10 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer, pipeline
 
-# Preload translation models
+# Load translation models for Urdu-English and English-Urdu
 model_names = {
     "en-ur": "Helsinki-NLP/opus-mt-en-ur",
-    "ur-en": "Helsinki-NLP/opus-mt-ur-en",
-    "en-fr": "Helsinki-NLP/opus-mt-en-fr",
-    "fr-en": "Helsinki-NLP/opus-mt-fr-en",
-    "en-de": "Helsinki-NLP/opus-mt-en-de",
-    "de-en": "Helsinki-NLP/opus-mt-de-en"
+    "ur-en": "Helsinki-NLP/opus-mt-ur-en"
 }
 
 models = {}
@@ -20,15 +16,15 @@ for key, model_name in model_names.items():
     models[key] = pipeline("translation", model=model, tokenizer=tokenizer)
 
 # Streamlit UI
-st.title("Language Translator")
+st.title("Urdu-English Translator")
 
 # User section on the left
 st.sidebar.title("User Section")
-input_lang = st.sidebar.selectbox("Select Input Language", ["English", "Urdu", "French", "Spanish", "German"])
-output_lang = st.sidebar.selectbox("Select Output Language", ["Urdu", "English", "French", "Spanish", "German"])
+input_lang = st.sidebar.selectbox("Select Input Language", ["English", "Urdu"])
+output_lang = st.sidebar.selectbox("Select Output Language", ["Urdu", "English"])
 
 # Text input
-input_text = st.text_area("Enter Text to Translate", placeholder = "Type here...")
+input_text = st.text_area("Enter Text to Translate",placehoder =  "Type here...")
 
 # Translation logic
 if st.button("Translate"):