Update app.py

app.py

@@ -50,7 +50,7 @@ def generate_image(prompt, model_name, steps, cfg_scale, width, height):
         raise gr.Error(f"Fehler bei '{model_name}': {str(e)}")
 
 # --- UI Aufbau ---
-with gr.Blocks(title="Custom Model Gen", theme=gr.themes.Base()) as demo:
+with gr.Blocks(title="Custom Model Gen") as demo:
     gr.Markdown("## ⚡ Custom Model Generator")
     
     with gr.Row():
@@ -109,4 +109,4 @@ with gr.Blocks(title="Custom Model Gen", theme=gr.themes.Base()) as demo:
     run_btn.click(fn=generate_image, inputs=inputs, outputs=result_output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(theme=gr.themes.Base())