Rodrigo_Cobo
commited on
Commit
·
eac489d
1
Parent(s):
4cb23f4
fix jpeg error
Browse files
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 |
|