Hugging Face's logo

Spaces:
ahmedxeno
/
depth_estimation
Runtime error

App Files Community
ahmedxeno commited on
Commit
b4bbe67
·
1 Parent(s): 9669154

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -29,8 +29,9 @@ def predict_image(img):
29
  ).squeeze()
30
 
31
  img = prediction.cpu().numpy()
 
32
 
33
- img = (img / 1000.0)*255
34
  out = (img).astype(np.uint8)
35
 
36
 
 
29
  ).squeeze()
30
 
31
  img = prediction.cpu().numpy()
32
+ a = img.max()
33
 
34
+ img = (img / a)*255
35
  out = (img).astype(np.uint8)
36
 
37