fffiloni commited on
Commit
3ce4287
1 Parent(s): 9d80522

Update demo_gradio.py

Browse files
Files changed (1) hide show
  1. 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)