Spaces:
Running
on
Zero
Running
on
Zero
hengli
commited on
Commit
·
acbafbf
1
Parent(s):
f84c11a
update hf link
Browse files- app.py +1 -1
- visual_util.py +0 -2
app.py
CHANGED
|
@@ -58,7 +58,7 @@ def run_model(target_dir, model, anchor_size=100) -> dict:
|
|
| 58 |
raise ValueError("CUDA is not available. Check your environment.")
|
| 59 |
|
| 60 |
# Move model to device
|
| 61 |
-
model = model.to(device)
|
| 62 |
model.eval()
|
| 63 |
|
| 64 |
# Load and preprocess images
|
|
|
|
| 58 |
raise ValueError("CUDA is not available. Check your environment.")
|
| 59 |
|
| 60 |
# Move model to device
|
| 61 |
+
# model = model.to(device)
|
| 62 |
model.eval()
|
| 63 |
|
| 64 |
# Load and preprocess images
|
visual_util.py
CHANGED
|
@@ -68,8 +68,6 @@ def predictions_to_glb(
|
|
| 68 |
except (ValueError, IndexError):
|
| 69 |
pass
|
| 70 |
|
| 71 |
-
for k,v in predictions.items():
|
| 72 |
-
print(f"{k} has {np.isnan(v).sum()} nan values")
|
| 73 |
if "Pointmap" in prediction_mode:
|
| 74 |
print("Using Pointmap Branch")
|
| 75 |
if "world_points" in predictions:
|
|
|
|
| 68 |
except (ValueError, IndexError):
|
| 69 |
pass
|
| 70 |
|
|
|
|
|
|
|
| 71 |
if "Pointmap" in prediction_mode:
|
| 72 |
print("Using Pointmap Branch")
|
| 73 |
if "world_points" in predictions:
|