Update visualization.py

visualization.py

@@ -215,7 +215,7 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, outpu
     out = cv2.VideoWriter(output_path, fourcc, fps, (width, height + 200))  # Additional 200 pixels for heatmap
     
     # Prepare the heatmap data
-    heatmap_data = np.vstack((mse_embeddings, mse_posture, mse_voice))
+    heatmap_data = np.vstack((mse_embeddings, mse_posture))
     
     # Create a figure for the heatmap
     fig, ax = plt.subplots(figsize=(width/100, 2))