Update visualization.py

visualization.py

@@ -278,8 +278,8 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     video = VideoFileClip(video_path)
     
     # Get video properties
-    width, height is video.w, video.h
-    total_frames is int(video.duration * video.fps)
+    width, height = video.w, video.h
+    total_frames = int(video.duration * video.fps)
     
     # Ensure all MSE arrays have the same length as total_frames
     mse_embeddings = np.interp(np.linspace(0, len(mse_embeddings) - 1, total_frames),