remove grid_query_frame

app.py

@@ -26,11 +26,9 @@ def parse_video(video_file):
 def cotracker_demo(
     input_video,
     grid_size: int = 10,
-    grid_query_frame: int = 0,
     tracks_leave_trace: bool = False,
 ):
     load_video = parse_video(input_video)
-    grid_query_frame = min(len(load_video) - 1, grid_query_frame)
     load_video = torch.from_numpy(load_video).permute(0, 3, 1, 2)[None].float()
 
     model = torch.hub.load("facebookresearch/co-tracker", "cotracker2_online")
@@ -39,7 +37,7 @@ def cotracker_demo(
         model = model.cuda()
         load_video = load_video.cuda()
 
-    model(video_chunk=load_video, is_first_step=True, grid_size=grid_size, grid_query_frame=grid_query_frame)
+    model(video_chunk=load_video, is_first_step=True, grid_size=grid_size)
     for ind in range(0, load_video.shape[1] - model.step, model.step):
         pred_tracks, pred_visibility = model(
             video_chunk=load_video[:, ind : ind + model.step * 2]
@@ -71,7 +69,6 @@ def cotracker_demo(
         tracks=pred_tracks.cpu(),
         visibility=pred_visibility.cpu(),
         filename=f"{filename}_pred_track",
-        query_frame=grid_query_frame,
     )
     return os.path.join(
         os.path.dirname(__file__), "results", f"{filename}_pred_track.mp4"
@@ -93,7 +90,6 @@ app = gr.Interface(
     <p> To get started, simply upload your <b>.mp4</b> video in landscape orientation or click on one of the example videos to load them. The shorter the video, the faster the processing. We recommend submitting short videos of length <b>2-7 seconds</b>.</p> \
     <ul style='display: inline-block; text-align: left;'> \
         <li>The total number of grid points is the square of <b>Grid Size</b>.</li> \
-        <li>To specify the starting frame for tracking, adjust <b>Grid Query Frame</b>. Tracks will be visualized only after the selected frame.</li> \
         <li>Check <b>Visualize Track Traces</b> to visualize traces of all the tracked points. </li> \
     </ul> \
     <p style='text-align: left'>For more details, check out our <a href='https://github.com/facebookresearch/co-tracker' target='_blank'>GitHub Repo</a> ⭐</p> \
@@ -102,16 +98,15 @@ app = gr.Interface(
     inputs=[
         gr.Video(type="file", label="Input video", interactive=True),
         gr.Slider(minimum=1, maximum=70, step=1, value=10, label="Grid Size"),
-        gr.Slider(minimum=0, maximum=30, step=1, default=0, label="Grid Query Frame"),
         gr.Checkbox(label="Visualize Track Traces"),
     ],
     outputs=gr.Video(label="Video with predicted tracks"),
     examples=[
-        [apple, 10, 0, False, False],
-        [apple, 20, 30, True, False],
-        [bear, 10, 0, False, False],
-        [paragliding, 10, 0, False, False],
-        [paragliding_launch, 10, 0, False, False],
+        [apple, 30, False],
+        [apple, 10, True],
+        [bear, 10, False],
+        [paragliding, 10, False],
+        [paragliding_launch, 10, False],
     ],
     cache_examples=True,
     allow_flagging=False,