Update app.py
Browse files
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])
|