nikkar commited on
Commit
d0b3c0b
·
verified ·
1 Parent(s): 0269ea8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)