Update visualization.py

visualization.py

@@ -229,7 +229,7 @@ def create_heatmap(frame_time, mse_embeddings, mse_posture, mse_voice):
     ax.barh(['Face', 'Posture', 'Voice'], mse_values, color=['navy', 'purple', 'green'])
     ax.set_xlim(0, 1)  # Normalize the MSE values
 
-    canvas = FigureCanvas(fig)
+    canvas = FigureCanvasAgg(fig)
     canvas.draw()
     img = np.frombuffer(canvas.tostring_rgb(), dtype='uint8')
     img = img.reshape(fig.canvas.get_width_height()[::-1] + (3,))