Update app.py
Browse files
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]
|