Update app.py

app.py

@@ -152,6 +152,8 @@ body {
 }
 """
 
+# --- Gradio Interface ---
+
 def start_generation():
     return demo
 
@@ -159,7 +161,7 @@ def start_generation():
 with gr.Blocks() as welcome_screen:
     with gr.Row():
         logo = gr.Image(
-            value="http://disneypixaraigenerator.com/wp-content/uploads/2023/12/cropped-android-chrome-512x512-1.png", # Replace with your logo URL
+            value="http://disneypixaraigenerator.com/wp-content/uploads/2023/12/cropped-android-chrome-512x512-1.png",
             elem_id="logo",
             height=200,
             width=300
@@ -171,6 +173,9 @@ with gr.Blocks() as welcome_screen:
     with gr.Row():
         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)
+
 # Main Generation Screen
 with gr.Blocks(css=css) as demo:
     with gr.Row():
@@ -224,8 +229,5 @@ 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)
 
-# Connect Welcome Screen to Generation Screen
-start_button.click(fn=start_generation, outputs=demo)
-
 # Launch the Gradio app
-welcome_screen.launch(max_threads=128) 
+welcome_screen.launch(max_threads=128)