Update app.py

app.py

@@ -154,6 +154,9 @@ body {
 
 # --- Gradio Interface ---
 
+# Define demo OUTSIDE the blocks
+demo = gr.Blocks(css=css) 
+
 def start_generation():
     return demo
 
@@ -176,8 +179,8 @@ with gr.Blocks() as welcome_screen:
     # Connect Welcome Screen to Generation Screen 
     start_button.click(fn=start_generation, outputs=demo)
 
-# Main Generation Screen
-with gr.Blocks(css=css) as demo:
+# Main Generation Screen (Now use the defined 'demo')
+with demo:
     with gr.Row():
         gr.Markdown("<h1 style='text-align: center;'>Create Your Disney Pixar AI Poster</h1>", elem_id="title")
 
@@ -230,4 +233,4 @@ with gr.Blocks(css=css) as demo:
     text_button.click(flip_text, inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed], outputs=image_output)
 
 # Launch the Gradio app
-welcome_screen.launch(max_threads=128)
+welcome_screen.launch(max_threads=128)