Update app.py

app.py

@@ -223,7 +223,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
     clip.write_videofile(video_file, codec='libx264', audio=False, verbose=False, logger=None)
     return video_file
 
-@spaces.GPU(duration=100)
+@spaces.GPU(duration=150)
 def track_video(n_frames, video_state):
 
     input_points = video_state["input_points"]
@@ -248,7 +248,9 @@ def track_video(n_frames, video_state):
     video_state["origin_images"] = images
     images = np.array(images)
 
-    video_predictor = video_predictor.to("cuda")
+    sam2_checkpoint = "./SAM2-Video-Predictor/checkpoints/sam2_hiera_large.pt"
+    config = "sam2_hiera_l.yaml"
+    video_predictor = build_sam2_video_predictor(config, sam2_checkpoint, device="cuda")
     
     inference_state = video_predictor.init_state(images=images/255, device="cuda")
     video_state["inference_state"] = inference_state