Spaces:
Sleeping
Sleeping
Update depth_app.py
Browse files- depth_app.py +2 -0
depth_app.py
CHANGED
@@ -98,6 +98,8 @@ def main():
|
|
98 |
model_path = '.' # Update this to the path of your model
|
99 |
encoder, depth_decoder, feed_height, feed_width = load_model(device, model_path)
|
100 |
|
|
|
|
|
101 |
colormapped_im = predict_depth(image, encoder, depth_decoder, device, feed_height, feed_width)
|
102 |
depth_image = pil.fromarray(colormapped_im)
|
103 |
|
|
|
98 |
model_path = '.' # Update this to the path of your model
|
99 |
encoder, depth_decoder, feed_height, feed_width = load_model(device, model_path)
|
100 |
|
101 |
+
encoder.eval()
|
102 |
+
depth_decoder.eval()
|
103 |
colormapped_im = predict_depth(image, encoder, depth_decoder, device, feed_height, feed_width)
|
104 |
depth_image = pil.fromarray(colormapped_im)
|
105 |
|