Update app.py

app.py

@@ -350,7 +350,7 @@ def track(
         query_points_tensor = query_points_tensor[None].flip(-1).to(device, dtype) # xyt -> tyx
         query_points_tensor = query_points_tensor[:, :, [0, 2, 1]] # tyx -> txy
 
-    video_input = torch.tensor(video_input).unsqueeze(0).to(device, dtype)
+    video_input = torch.tensor(video_input).unsqueeze(0)
 
     model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
     model = model.to(device)
@@ -381,7 +381,7 @@ def track(
     # 
     for ind in range(0, video_input.shape[1] - model.step, model.step):
         pred_tracks, pred_visibility = model(
-            video_chunk=video_input[:, ind : ind + model.step * 2],
+            video_chunk=video_input[:, ind : ind + model.step * 2].to(device, dtype),
             grid_size=0, 
             queries=queries, 
             add_support_grid=add_support_grid