bartduis commited on
Commit
32d1162
·
verified ·
1 Parent(s): 736cf6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -111,7 +111,7 @@ def rayst3r_to_glb(img,depth_dict,mask,max_total_points=10e6,rotated=False):
111
  rayst3r_model = EvalWrapper(rayst3r_checkpoint,device='cpu')
112
  rayst3r_model = rayst3r_model.to(device)
113
 
114
- rayst3r_points = eval_scene(rayst3r_model,os.path.join(outdir, "input"),do_filter_all_masks=True,dino_model=dino_model).cpu()
115
 
116
  # subsample points
117
  n_points = min(max_total_points,rayst3r_points.shape[0])
 
111
  rayst3r_model = EvalWrapper(rayst3r_checkpoint,device='cpu')
112
  rayst3r_model = rayst3r_model.to(device)
113
 
114
+ rayst3r_points = eval_scene(rayst3r_model,os.path.join(outdir, "input"),do_filter_all_masks=True,dino_model=dino_model, device = device).cpu()
115
 
116
  # subsample points
117
  n_points = min(max_total_points,rayst3r_points.shape[0])