Update visualization.py

visualization.py

@@ -226,7 +226,7 @@ def fill_with_zeros(mse_array, total_frames):
     return result
 
 def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, desired_fps, total_frames, video_width):
-    fig, ax = plt.subplots(figsize=(video_width / 100, 0.4))
+    fig, ax = plt.subplots(figsize=(video_width / 250, 0.4))
 
     # Create the full heatmap for the entire video duration
     combined_mse = np.array([mse_embeddings, mse_posture, mse_voice])