Update app.py

app.py

@@ -394,7 +394,7 @@ def track(
     pred_tracks = (pred_tracks * torch.tensor([video_preview.shape[2], video_preview.shape[1]]).to(device) / torch.tensor([VIDEO_INPUT_RESO[1], VIDEO_INPUT_RESO[0]]).to(device))
     
     vis = Visualizer(save_dir="./saved_videos", pad_value=120, linewidth=3)
-    painted_video = vis.visualize(torch.tensor(video_preview.permute(0, 3, 1, 2)[None]).to(pred_tracks.device), pred_tracks, pred_visibility, save_video=False)[0].permute(0, 2, 3, 1).cpu().numpy()
+    painted_video = vis.visualize(torch.tensor(video_preview).permute(0, 3, 1, 2)[None].to(pred_tracks.device), pred_tracks, pred_visibility, save_video=False)[0].permute(0, 2, 3, 1).cpu().numpy()
     
     # painted_video = paint_point_track(video_preview,tracks,pred_occ,colors)