nikkar commited on
Commit
2e68677
·
verified ·
1 Parent(s): d0b6f98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -482,7 +482,7 @@ def track(
482
  torch.tensor(video_preview)
483
  .permute(0, 3, 1, 2)[None]
484
  .to(pred_tracks.device),
485
- pred_tracks,
486
  pred_visibility,
487
  save_video=False,
488
  )[0]
 
482
  torch.tensor(video_preview)
483
  .permute(0, 3, 1, 2)[None]
484
  .to(pred_tracks.device),
485
+ pred_tracks.round(),
486
  pred_visibility,
487
  save_video=False,
488
  )[0]