Update app.py

app.py

@@ -18,6 +18,7 @@ from typing import List, Optional, Sequence, Tuple
 import spaces
 import numpy as np
 from visualizer import Visualizer
+from predictor import CoTrackerOnlinePredictor
 
 
 # Generate random colormaps for visualizing different points.
@@ -404,7 +405,8 @@ def track(
 
     video_input = torch.tensor(video_input).unsqueeze(0)
 
-    model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
+    # model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
+    model = CoTrackerOnlinePredictor()
     model = model.to(device)
 
     video_input = video_input.permute(0, 1, 4, 2, 3)