Update app.py

app.py

@@ -1,14 +1,23 @@
 import os
 os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
 
-from transformers import MBart50TokenizerFast, MBartForConditionalGeneration
+from transformers import (
+    MBart50TokenizerFast,
+    MBartForConditionalGeneration,
+    AutoConfig
+)
 import gradio as gr
 
 # ---- Load model and tokenizer ----
 model_name = "Mudasir692/mbart-eng-ur"
 
+# Fix config issue (early_stopping = None)
+config = AutoConfig.from_pretrained(model_name)
+if getattr(config, "early_stopping", None) is None:
+    config.early_stopping = True
+
 tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX", tgt_lang="ur_PK")
-model = MBartForConditionalGeneration.from_pretrained(model_name)
+model = MBartForConditionalGeneration.from_pretrained(model_name, config=config)
 
 # ---- Translation function ----
 def translate_to_urdu(text):
@@ -44,12 +53,17 @@ app = gr.Interface(
     theme="soft",
     css="""
         body {
-            background: linear-gradient(to bottom right, #dff9fb, #c7ecee);
+            background: linear-gradient(135deg, #d1f2eb, #a3e4d7, #d6eaf8);
             font-family: 'Segoe UI', sans-serif;
+            background-attachment: fixed;
         }
         .gr-button-primary {
             background-color: #1e3799 !important;
             color: white !important;
+            border-radius: 8px !important;
+        }
+        h1, h3 {
+            color: #0a3d62 !important;
         }
     """,
 )