Spaces:
Running
on
Zero
Running
on
Zero
put image on correct device
Browse files
app.py
CHANGED
|
@@ -71,6 +71,7 @@ def run_rerun(path_to_video):
|
|
| 71 |
yield stream.read()
|
| 72 |
|
| 73 |
image = transform(frame)
|
|
|
|
| 74 |
depth, focal_length = estimate_depth(image)
|
| 75 |
|
| 76 |
rr.log(
|
|
|
|
| 71 |
yield stream.read()
|
| 72 |
|
| 73 |
image = transform(frame)
|
| 74 |
+
image = image.to(device)
|
| 75 |
depth, focal_length = estimate_depth(image)
|
| 76 |
|
| 77 |
rr.log(
|