Spaces:
Running
on
Zero
Running
on
Zero
slothfulxtx
commited on
Commit
·
e2c768c
1
Parent(s):
d8b1630
fix path bug
Browse files
app.py
CHANGED
|
@@ -167,6 +167,8 @@ def infer_geometry(
|
|
| 167 |
point_maps = point_maps.cpu()
|
| 168 |
valid_masks = valid_masks.cpu()
|
| 169 |
|
|
|
|
|
|
|
| 170 |
gc.collect()
|
| 171 |
torch.cuda.empty_cache()
|
| 172 |
output_npz_path = Path(save_folder, run_id, f'point_maps.npz')
|
|
|
|
| 167 |
point_maps = point_maps.cpu()
|
| 168 |
valid_masks = valid_masks.cpu()
|
| 169 |
|
| 170 |
+
os.makedirs(save_folder, exist_ok=True)
|
| 171 |
+
|
| 172 |
gc.collect()
|
| 173 |
torch.cuda.empty_cache()
|
| 174 |
output_npz_path = Path(save_folder, run_id, f'point_maps.npz')
|