Update app.py
Browse files
app.py
CHANGED
@@ -406,7 +406,7 @@ def track(
|
|
406 |
video_input = torch.tensor(video_input).unsqueeze(0)
|
407 |
|
408 |
# model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
|
409 |
-
model = CoTrackerOnlinePredictor()
|
410 |
model = model.to(device)
|
411 |
|
412 |
video_input = video_input.permute(0, 1, 4, 2, 3)
|
|
|
406 |
video_input = torch.tensor(video_input).unsqueeze(0)
|
407 |
|
408 |
# model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
|
409 |
+
model = CoTrackerOnlinePredictor(checkpoint='./baseline_online.pth')
|
410 |
model = model.to(device)
|
411 |
|
412 |
video_input = video_input.permute(0, 1, 4, 2, 3)
|