Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ def process_image(image_path):
|
|
43 |
output = prediction.cpu().numpy()
|
44 |
depth_image = (output * 255 / np.max(output)).astype("uint8")
|
45 |
depth_image_padded = np.array(
|
46 |
-
|
47 |
)
|
48 |
|
49 |
|
|
|
43 |
output = prediction.cpu().numpy()
|
44 |
depth_image = (output * 255 / np.max(output)).astype("uint8")
|
45 |
depth_image_padded = np.array(
|
46 |
+
Image.fromarray(depth_image)
|
47 |
)
|
48 |
|
49 |
|