Update app.py

app.py

@@ -1,13 +1,13 @@
 import gradio as gr
 from transformers import MarianMTModel, MarianTokenizer
 
-# Define model names for each direction
+# Define model names
 model_names = {
     "English to Urdu": "Helsinki-NLP/opus-mt-en-ur",
     "Urdu to English": "Helsinki-NLP/opus-mt-ur-en"
 }
 
-# Cache loaded models
+# Load models only once
 models = {}
 
 def load_model(direction):
@@ -20,21 +20,22 @@ def load_model(direction):
 
 def translate(text, direction):
     if not text.strip():
-        return ""
+        return "Please enter some text to translate."
     tokenizer, model = load_model(direction)
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
-    translated = model.generate(**inputs)
-    return tokenizer.decode(translated[0], skip_special_tokens=True)
+    outputs = model.generate(**inputs)
+    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return translated_text
 
 iface = gr.Interface(
     fn=translate,
     inputs=[
-        gr.Textbox(label="Enter Text", lines=4),
-        gr.Radio(["English to Urdu", "Urdu to English"], label="Translation Direction")
+        gr.Textbox(label="Enter Text", lines=4, placeholder="Type here..."),
+        gr.Radio(["English to Urdu", "Urdu to English"], label="Select Direction")
     ],
     outputs=gr.Textbox(label="Translated Text"),
     title="English ↔ Urdu Translator",
-    description="Translate between English and Urdu using pretrained Hugging Face models.",
+    description="Use this app to translate between English and Urdu using Hugging Face MarianMT models.",
     allow_flagging="never"
 )