Update app.py

app.py

@@ -3,7 +3,7 @@ from gradio_client import Client, handle_file
 import seaborn as sns
 import matplotlib.pyplot as plt
 import os
-
+import pandas as pd
 
 # Define your Hugging Face token (make sure to set it as an environment variable)
 HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using an environment variable
@@ -47,9 +47,11 @@ def heatmap(top_n):
     )
     print(pivot_table)
     print(type(pivot_table))
+
+    df = pd.read_csv(StringIO(pivot_table), delim_whitespace=True, index_col=0)
     
     plt.figure(figsize=(10, 8))
-    sns.heatmap(pivot_table,
+    sns.heatmap(df,
                 cmap='YlOrRd',
                 cbar_kws={'label': 'Weighted Frequency'},
                 square=True)