Rodrigo_Cobo commited on
Commit
eac489d
1 Parent(s): 4cb23f4

fix jpeg error

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -47,8 +47,10 @@ def update(slider, img):
47
 
48
  output = prediction.cpu().numpy()
49
 
50
- out_im = Image.fromarray(output)
51
- out_im.convert('RGB').save("temp/image_depth.jpeg", "JPEG")
 
 
52
 
53
  return f'temp/image_depth.jpeg'
54
 
 
47
 
48
  output = prediction.cpu().numpy()
49
 
50
+ #out_im = Image.fromarray(output)
51
+ #out_im.convert('RGB').save("temp/image_depth.jpeg", "JPEG")
52
+
53
+ cv2.imwrite("temp/image_depth.jpeg", output)
54
 
55
  return f'temp/image_depth.jpeg'
56