zino36 commited on
Commit
5e51d3c
·
verified ·
1 Parent(s): 7798364

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -82,9 +82,9 @@ with gr.Blocks(css=css) as demo:
82
  original_image = image.copy()
83
 
84
 
85
- depth = predict_depth(image)
86
 
87
- return [(original_image, depth)]
88
 
89
  submit.click(on_submit, inputs=[input_image], outputs=[depth_image_slider, gray_depth_file, raw_file])
90
 
 
82
  original_image = image.copy()
83
 
84
 
85
+ result = predict_depth(image)
86
 
87
+ return [(original_image, result.depth)]
88
 
89
  submit.click(on_submit, inputs=[input_image], outputs=[depth_image_slider, gray_depth_file, raw_file])
90