Rodrigo_Cobo commited on
Commit
6a7e68e
1 Parent(s): 61d09e6

fix jpeg error

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