Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -351,7 +351,7 @@ def track(
|
|
351 |
|
352 |
video_input = torch.tensor(video_input).unsqueeze(0).to(device, dtype)
|
353 |
|
354 |
-
model = torch.hub.load("facebookresearch/co-tracker
|
355 |
model = model.to(device)
|
356 |
|
357 |
video_input = video_input.permute(0, 1, 4, 2, 3)
|
|
|
351 |
|
352 |
video_input = torch.tensor(video_input).unsqueeze(0).to(device, dtype)
|
353 |
|
354 |
+
model = torch.hub.load("facebookresearch/co-tracker", "cotracker3_online")
|
355 |
model = model.to(device)
|
356 |
|
357 |
video_input = video_input.permute(0, 1, 4, 2, 3)
|