Update app.py

app.py

@@ -221,8 +221,8 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
     return video_file
 
 @spaces.GPU(duration=60)
-def track_video(n_frames, video_state):
-
+def track_video(n_frames):
+    global video_state
     input_points = video_state["input_points"]
     input_labels = video_state["input_labels"]
     frame_idx = video_state["frame_idx"]
@@ -407,6 +407,6 @@ with gr.Blocks() as demo:
                        outputs=image_output)
         image_output.select(fn=segment_frame, inputs=[point_prompt, video_state], outputs=image_output)
         clear_btn.click(clear_clicks, inputs=video_state, outputs=image_output)
-        track_btn.click(track_video, inputs=[n_frames_slider, video_state], outputs=video_output)
+        track_btn.click(track_video, inputs=[n_frames_slider], outputs=video_output)
 
 demo.launch()