Update app.py

app.py

@@ -18,9 +18,9 @@ def run_theia(image):
     ]
     feature_means, feature_vars = load_feature_stats(target_model_names, stat_file_root="feature_stats")
     
-    mask_generator, sam_model = prepare_mask_generator(device)
+    mask_generator, sam_model = prepare_mask_generator('cuda')
     depth_anything_model_name = "LiheYoung/depth-anything-large-hf"
-    depth_anything_decoder, _ = prepare_depth_decoder(depth_anything_model_name, device)
+    depth_anything_decoder, _ = prepare_depth_decoder(depth_anything_model_name, 'cuda')
     
     images = [image]
     
@@ -35,7 +35,7 @@ def run_theia(image):
         pred_iou_thresh=0.5,
         stability_score_thresh=0.7,
         gt=True,
-        device=device,
+        device='cuda',
     )
     
     vis_video = np.stack(