Hugging Face's logo

Spaces:
NN-BRD
/
hackathon_depth_segment
Runtime error

App Files Community
jens commited on
Commit
62a1e0a
·
1 Parent(s): 7598e8a

fix

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -33,8 +33,8 @@ class DepthPredictor:
33
 
34
  output = prediction.cpu().numpy()
35
  formatted = (output * 255 / np.max(output)).astype('uint8')
36
- img = Image.fromarray(formatted)
37
- return img
38
 
39
 
40
 
 
33
 
34
  output = prediction.cpu().numpy()
35
  formatted = (output * 255 / np.max(output)).astype('uint8')
36
+ #img = Image.fromarray(formatted)
37
+ return formatted
38
 
39
 
40