Update app.py

app.py

@@ -248,8 +248,7 @@ def track_video(n_frames, video_state):
     video_state["origin_images"] = images
     images = np.array(images)
 
-    video_predictor = build_sam2_video_predictor(config, sam2_checkpoint, device="cuda")
-
+    video_predictor = video_predictor.to("cuda")
     
     inference_state = video_predictor.init_state(images=images/255, device="cuda")
     video_state["inference_state"] = inference_state