Update visualization.py

visualization.py

@@ -268,7 +268,8 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     def create_heatmap(t):
         frame_count = int(t * video.fps)
         
-        if hasattr(ax, 'line'):
+        # Clear previous lines
+        if hasattr(ax, 'lines') and len(ax.lines) > 0:
             ax.lines.pop(0)
         ax.axvline(x=frame_count, color='blue', linewidth=2)