Update app.py

app.py

@@ -274,7 +274,10 @@ def train_model(dataset_file: gr.File, batch_size: int, epochs: int, lr: float,
 
 
 # --- Gradio Interface Setup (using Blocks for a nicer layout) ---
-with gr.Blocks(title="LayoutLMv3 Fine-Tuning using MLP instead of Linear Classifier", theme="soft") as demo:
+# Removing title and theme from here prevents the __init__ crash
+with gr.Blocks() as demo:
+    gr.Markdown("# 🚀 LayoutLMv3 Fine-Tuning: MLP vs Linear Classifier")
+    gr.Markdown("### Training App by Aastik")
     gr.Markdown("# 🚀 LayoutLMv3 Fine-Tuning on Hugging Face Spaces")
     gr.Markdown(
         """