Update demo_gradio.py

demo_gradio.py

@@ -164,19 +164,26 @@ def greet(input_image, material_exemplar):
     
     return images[0]
 
+css = """
+#col-container{
+    margin: 0 auto;
+    max-width: 1080px;
+}
+"""
 
-
-input_image = gr.Image(type="pil")
-input_image2 = gr.Image(type="pil")
-
-demo = gr.Interface(
-    fn=greet,
-    inputs=[input_image, input_image2],
-    title="ZeST: Zero-Shot Material Transfer from a Single Image",
-    description="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.",
-    outputs=["image"],
-    allow_flagging='never'
-
-)
-
-demo.launch()
+with gr.Blocks(css=css) as demo:
+    with gr.Column(elem_id="col-container"):
+        gr.Markdown("""
+        # 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():
+            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()
+    submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
+
+demo.queue().launch()