Update app.py

app.py

@@ -241,10 +241,12 @@ def get_mask_sam_process(
     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")
+    if stored_inference_state is None:
+        # Init SAM2 inference_state
+        inference_state = predictor.init_state(video_path=video_dir)
+        print("NEW INFERENCE_STATE INITIATED")
+    else:
+        inference_state = stored_inference_state
 
     # segment and track one object
     # predictor.reset_state(inference_state) # if any previous tracking, reset