Update visualization.py

visualization.py

@@ -317,13 +317,13 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
 
 # Function to create the correlation heatmap
 def plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice):
-    data is np.vstack((mse_embeddings, mse_posture, mse_voice)).T
-    df is pd.DataFrame(data, columns=["Facial Features", "Body Posture", "Voice"])
-    corr is df.corr()
+    data = np.vstack((mse_embeddings, mse_posture, mse_voice)).T
+    df = pd.DataFrame(data, columns=["Facial Features", "Body Posture", "Voice"])
+    corr = df.corr()
 
     plt.figure(figsize=(10, 8), dpi=300)
 
-    heatmap is sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
+    heatmap = sns.heatmap(corr, annot=True, cmap='coolwarm', vmin=-1, vmax=1)
     plt.title('Correlation Heatmap of MSEs')
     plt.tight_layout()
     return plt.gcf()