Update app.py

app.py

@@ -28,7 +28,7 @@ from PIL import Image, ImageFilter
 from sam2.build_sam import build_sam2_video_predictor
 
 def preprocess_image(image):
-    return image, gr.State([]), gr.State([]), image, gr.State([])
+    return image, gr.State([]), gr.State([]), image, gr.State()
 
 def preprocess_video_in(video_path):
 
@@ -70,7 +70,7 @@ def preprocess_video_in(video_path):
     cap.release()
 
     # 'image' is the first frame extracted from video_in
-    return first_frame, gr.State([]), gr.State([]), first_frame, first_frame, output_dir, gr.State([]), gr.State([])
+    return first_frame, gr.State([]), gr.State([]), first_frame, first_frame, output_dir, gr.State(), gr.State()
 
 def get_point(point_type, tracking_points, trackings_input_label, first_frame_path, evt: gr.SelectData):
     print(f"You selected {evt.value} at {evt.index} from {evt.target}")
@@ -304,8 +304,8 @@ with gr.Blocks() as demo:
     tracking_points = gr.State([])
     trackings_input_label = gr.State([])
     video_frames_dir = gr.State()
-    stored_inference_state = gr.State([])
-    stored_frame_names = gr.State([])
+    stored_inference_state = gr.State()
+    stored_frame_names = gr.State()
     with gr.Column():
         gr.Markdown("# SAM2 Video Predictor")
         gr.Markdown("This is a simple demo for video segmentation with SAM2.")