Spaces:
Runtime error
Runtime error
Commit
·
163beec
1
Parent(s):
a36a6fa
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,9 +14,10 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 14 |
zoe = model_zoe_n.to(DEVICE)
|
| 15 |
|
| 16 |
def process_image(image):
|
|
|
|
| 17 |
depth = zoe.infer_pil(image) # as numpy
|
| 18 |
-
|
| 19 |
colored_depth = colorize(depth)
|
|
|
|
| 20 |
|
| 21 |
return colored_depth
|
| 22 |
|
|
|
|
| 14 |
zoe = model_zoe_n.to(DEVICE)
|
| 15 |
|
| 16 |
def process_image(image):
|
| 17 |
+
|
| 18 |
depth = zoe.infer_pil(image) # as numpy
|
|
|
|
| 19 |
colored_depth = colorize(depth)
|
| 20 |
+
colored_depth = colored_depth.resize((512,512))
|
| 21 |
|
| 22 |
return colored_depth
|
| 23 |
|