Update app.py

app.py

@@ -177,7 +177,7 @@ with gr.Blocks() as welcome_screen:
         start_button = gr.Button("Get Started", variant='primary', elem_id="start-button")
 
     # Connect Welcome Screen to Generation Screen 
-    start_button.click(fn=start_generation, outputs=demo)
+    start_button.click(fn=start_generation, outputs=welcome_screen) # Output welcome_screen
 
 # Main Generation Screen (Now use the defined 'demo')
 with demo:
@@ -233,4 +233,4 @@ with 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)