Spaces:
Running
on
Zero
Running
on
Zero
fix inference
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def run_model(target_dir, model) -> dict:
|
|
| 77 |
|
| 78 |
with torch.no_grad():
|
| 79 |
with torch.cuda.amp.autocast(dtype=dtype):
|
| 80 |
-
predictions = model(images)
|
| 81 |
|
| 82 |
# Convert pose encoding to extrinsic and intrinsic matrices
|
| 83 |
print("Converting pose encoding to extrinsic and intrinsic matrices...")
|
|
|
|
| 77 |
|
| 78 |
with torch.no_grad():
|
| 79 |
with torch.cuda.amp.autocast(dtype=dtype):
|
| 80 |
+
predictions = model.inference(images)
|
| 81 |
|
| 82 |
# Convert pose encoding to extrinsic and intrinsic matrices
|
| 83 |
print("Converting pose encoding to extrinsic and intrinsic matrices...")
|