Update demo_gradio.py
Browse files- demo_gradio.py +8 -6
demo_gradio.py
CHANGED
@@ -167,7 +167,7 @@ def greet(input_image, material_exemplar):
|
|
167 |
css = """
|
168 |
#col-container{
|
169 |
margin: 0 auto;
|
170 |
-
max-width:
|
171 |
}
|
172 |
"""
|
173 |
|
@@ -177,11 +177,13 @@ with gr.Blocks(css=css) as demo:
|
|
177 |
# ZeST: Zero-Shot Material Transfer from a Single Image
|
178 |
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.
|
179 |
""")
|
180 |
-
with gr.
|
181 |
-
|
182 |
-
|
183 |
-
|
184 |
-
|
|
|
|
|
185 |
submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
|
186 |
|
187 |
demo.queue().launch()
|
|
|
167 |
css = """
|
168 |
#col-container{
|
169 |
margin: 0 auto;
|
170 |
+
max-width: 960px;
|
171 |
}
|
172 |
"""
|
173 |
|
|
|
177 |
# ZeST: Zero-Shot Material Transfer from a Single Image
|
178 |
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.
|
179 |
""")
|
180 |
+
with gr.Column():
|
181 |
+
with gr.Row():
|
182 |
+
input_image = gr.Image(type="pil")
|
183 |
+
input_image2 = gr.Image(type="pil")
|
184 |
+
submit_btn = gr.Button("Submit")
|
185 |
+
with gr.Column():
|
186 |
+
output_image = gr.Image()
|
187 |
submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
|
188 |
|
189 |
demo.queue().launch()
|