Update demo_gradio.py

demo_gradio.py

@@ -167,7 +167,7 @@ def greet(input_image, material_exemplar):
 css = """
 #col-container{
     margin: 0 auto;
-    max-width: 640px;
+    max-width: 960px;
 }
 """
 
@@ -177,11 +177,13 @@ with gr.Blocks(css=css) as demo:
         # ZeST: Zero-Shot Material Transfer from a Single Image
         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():
-            input_image = gr.Image(type="pil")
-            input_image2 = gr.Image(type="pil")
-        submit_btn = gr.Button("Submit")
-        output_image = gr.Image()
+        with gr.Column():
+            with gr.Row():
+                input_image = gr.Image(type="pil")
+                input_image2 = gr.Image(type="pil")
+            submit_btn = gr.Button("Submit")
+        with gr.Column():
+            output_image = gr.Image()
     submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
 
 demo.queue().launch()