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 |
|