mkalia commited on
Commit
9ca77c2
·
verified ·
1 Parent(s): d7357cf

Update depth_app.py

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