Update visualization.py

visualization.py

@@ -203,6 +203,7 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
 
 def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, output_path):
     from matplotlib.backends.backend_agg import FigureCanvasAgg
+    import cv2
     # Open the video
     cap = cv2.VideoCapture(video_path)
     fps = cap.get(cv2.CAP_PROP_FPS)