Update app.py

app.py

@@ -57,8 +57,6 @@ with gr.Blocks() as demo:
     # text_input = gr.Textbox(label="Enter prompt")
     # model_select = gr.Dropdown(label="Select Model", choices=['sd1', 'sd2'])
     # seed_input = gr.Number(label="Enter Seed", value=0)  # Default seed set to 0
-    with gr.Column():
-        gallery = gr.Gallery(label="Generated Images", columns=4, rows=3, object_fit="contain", height="auto")
     # Update the submit function to include the new inputs
 
     
@@ -96,6 +94,9 @@ with gr.Blocks() as demo:
 
     generate_button = gr.Button("Generate Image")
 
+    with gr.Column():
+        gallery = gr.Gallery(label="Generated Images", columns=4, rows=3, object_fit="contain", height="auto")
+
     gr.on(
         triggers=[
             # prompt.submit,