Rodrigo_Cobo
commited on
Commit
·
4cb23f4
1
Parent(s):
6a7e68e
fix jpeg error
Browse files
app.py
CHANGED
|
@@ -48,7 +48,7 @@ def update(slider, img):
|
|
| 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 |
|
|
|
|
| 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 |
|