Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def process_image(image):
|
|
31 |
output = prediction.cpu().numpy()
|
32 |
formatted = (output * 255 / np.max(output)).astype('uint8')
|
33 |
img = Image.fromarray(formatted)
|
34 |
-
return img
|
35 |
|
36 |
# Interface
|
37 |
title = "Demo: Zero-shot Depth Estimation with DPT"
|
|
|
31 |
output = prediction.cpu().numpy()
|
32 |
formatted = (output * 255 / np.max(output)).astype('uint8')
|
33 |
img = Image.fromarray(formatted)
|
34 |
+
return [img]
|
35 |
|
36 |
# Interface
|
37 |
title = "Demo: Zero-shot Depth Estimation with DPT"
|