hengli commited on
Commit
acbafbf
·
1 Parent(s): f84c11a

update hf link

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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: