Update demo_gradio.py

demo_gradio.py

@@ -178,12 +178,10 @@ with gr.Blocks(css=css) as demo:
         Upload two images -- input image and material exemplar. ZeST extracts the material from the exemplar and cast it onto the input image following the original lighting cues.
         """)
         with gr.Row():
-            with gr.Column():
-                with gr.Row():
-                    input_image = gr.Image(type="pil")
-                    input_image2 = gr.Image(type="pil")
-                submit_btn = gr.Button("Submit")
-            output_image = gr.Image()
+            input_image = gr.Image(type="pil")
+            input_image2 = gr.Image(type="pil")
+        submit_btn = gr.Button("Submit")
+        output_image = gr.Image()
     submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
 
 demo.queue().launch()