Spaces:
Running
on
Zero
Running
on
Zero
Update demo_gradio.py
Browse files- demo_gradio.py +2 -2
demo_gradio.py
CHANGED
@@ -91,7 +91,7 @@ transform = Compose(
|
|
91 |
model.eval()
|
92 |
|
93 |
@spaces.GPU()
|
94 |
-
def greet(input_image, material_exemplar):
|
95 |
|
96 |
"""
|
97 |
Compute depth map from input_image
|
@@ -193,4 +193,4 @@ with gr.Blocks(css=css) as demo:
|
|
193 |
output_image = gr.Image(label="transfer result")
|
194 |
submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
|
195 |
|
196 |
-
demo.queue().launch()
|
|
|
91 |
model.eval()
|
92 |
|
93 |
@spaces.GPU()
|
94 |
+
def greet(input_image, material_exemplar, gr.Progress(track_tqdm=True)):
|
95 |
|
96 |
"""
|
97 |
Compute depth map from input_image
|
|
|
193 |
output_image = gr.Image(label="transfer result")
|
194 |
submit_btn.click(fn=greet, inputs=[input_image, input_image2], outputs=[output_image])
|
195 |
|
196 |
+
demo.queue().launch(show_error=True)
|