Update visualization.py

visualization.py

@@ -230,10 +230,10 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
     combined_mse[1] = mse_posture_norm
     combined_mse[2] = mse_voice_norm
 
-    fig, ax = plt.subplots(figsize=(video_width / 100, 2))
+    fig, ax = plt.subplots(figsize=(video_width / 100, 1))  # Adjusted height to 1 for less vertical space
     ax.imshow(combined_mse, aspect='auto', cmap='coolwarm', vmin=0, vmax=1, extent=[0, total_frames, 0, 3])
     ax.set_yticks([0.5, 1.5, 2.5])
-    ax.set_yticklabels(['Face', 'Posture', 'Voice'])
+    ax.set_yticklabels(['Voice', 'Posture', 'Face'])
     ax.set_xticks([])
 
     ax.axvline(x=frame_count, color='blue', linewidth=2)