Update app.py

app.py

@@ -212,6 +212,7 @@ def load_model(checkpoint):
     
     
 def get_mask_sam_process(
+    stored_inference_state,
     input_first_frame_image, 
     checkpoint, 
     tracking_points, 
@@ -239,6 +240,8 @@ def get_mask_sam_process(
     # scan all the JPEG frame names in this directory
     frame_names = scanned_frames
 
+    print(f"STORED INFERENCE STEP: {stored_inference_state}")
+    # if stored_inference_state is None:
     # Init SAM2 inference_state
     inference_state = predictor.init_state(video_path=video_dir)
     print("NEW INFERENCE_STATE INITIATED")
@@ -506,6 +509,7 @@ with gr.Blocks() as demo:
     submit_btn.click(
         fn = get_mask_sam_process,
         inputs = [
+            stored_inference_state,
             input_first_frame_image, 
             checkpoint, 
             tracking_points,