Update app.py

app.py

@@ -51,10 +51,10 @@ def heatmap(top_n):
     print(pivot_table)
     print(type(pivot_table))
 
-    df = pd.read_csv(StringIO(pivot_table), delim_whitespace=True, index_col=0)
+    # df = pd.read_csv(StringIO(pivot_table), delim_whitespace=True, index_col=0)
     
     plt.figure(figsize=(10, 8))
-    sns.heatmap(df,
+    sns.heatmap(pivot_table,
                 cmap='YlOrRd',
                 cbar_kws={'label': 'Weighted Frequency'},
                 square=True)