Update app.py

app.py

@@ -143,8 +143,7 @@ with gr.Blocks(css='style.css') as demo:
             </div>
         """
     )
-    with gr.Blocks() as demo:
-    with gr.Column(elem_id="col-container"):
+        with gr.Column(elem_id="col-container"):
         with gr.Row(variant="compact"):
             input_text = gr.Textbox(
                 label="Short Prompt",
@@ -169,7 +168,8 @@ with gr.Blocks(css='style.css') as demo:
             run = gr.Button("Generate Images").style(full_width=False)
         with gr.Row():
             with gr.Row():
-               image_input = gr.Image()
+                with gr.Blocks() as demo:
+                    image_input = gr.Image()
         # image_output = gr.Image() # download [ok]
         image_output = gr.Gallery(
                 label='Output',