Update app.py

app.py

@@ -392,7 +392,7 @@ def track(
     colors = np.array(colors)
     
     vis = Visualizer(save_dir="./saved_videos", pad_value=120, linewidth=3)
-    painted_video = vis.visualize(video_preview, tracks, pred_occ, save_video=False)
+    painted_video = vis.visualize(video_input, 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)