Update app.py

app.py

@@ -2,17 +2,23 @@ import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer, pipeline
 
 # Preload translation models
-models = {
-    "en-ur": pipeline("translation", model=MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-en-ur"),
-                      tokenizer=MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-en-ur")),
-    "ur-en": pipeline("translation", model=MarianMTModel.from_pretrained("Helsinki-NLP/opus-mt-ur-en"),
-                      tokenizer=MarianTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ur-en")),
-    "en-fr": pipeline("translation_en_to_fr"),
-    "fr-en": pipeline("translation_fr_to_en"),
-    "en-de": pipeline("translation_en_to_de"),
-    "de-en": pipeline("translation_de_to_en")
+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"
 }
 
+models = {}
+
+# Load models once at the start
+for key, model_name in model_names.items():
+    tokenizer = MarianTokenizer.from_pretrained(model_name)
+    model = MarianMTModel.from_pretrained(model_name)
+    models[key] = pipeline("translation", model=model, tokenizer=tokenizer)
+
 # Streamlit UI
 st.title("Language Translator")
 
@@ -22,10 +28,9 @@ input_lang = st.sidebar.selectbox("Select Input Language", ["English", "Urdu", "
 output_lang = st.sidebar.selectbox("Select Output Language", ["Urdu", "English", "French", "Spanish", "German"])
 
 # Text input
-input_text = st.text_area("Enter Text to Translate",placeholder =  "Type here...")
+input_text = st.text_area("Enter Text to Translate", placeholder = "Type here...")
 
 # Translation logic
-translator = None
 if st.button("Translate"):
     model_key = f"{input_lang.lower()[0:2]}-{output_lang.lower()[0:2]}"
     if model_key in models: