Update demo_gradio.py

demo_gradio.py

@@ -177,13 +177,14 @@ 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.Column():
-            with gr.Row():
-                input_image = gr.Image(type="pil")
-                input_image2 = gr.Image(type="pil")
+        with gr.Row():
+            with gr.Column():
+                with gr.Row():
+                    input_image = gr.Image(type="pil", label="input image")
+                    input_image2 = gr.Image(type="pil", label = "material examplar")
             submit_btn = gr.Button("Submit")
-        with gr.Column():
-            output_image = gr.Image()
+            with gr.Column():
+                output_image = gr.Image(label="transfer result")
     submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
 
 demo.queue().launch()