vickeee465 commited on
Commit
5250894
·
1 Parent(s): 74fed9b

variable name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -94,7 +94,7 @@ def prepare_heatmap_data(data):
94
  heatmap_data.index = [item["sentence"] for item in data]
95
  return heatmap_data
96
 
97
- def plot_emotion_heatmap(data):
98
  fig = plt.figure(figsize=(10, len(data) * 0.5 + 2))
99
  sns.heatmap(heatmap_data, annot=True, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
100
  plt.title("Emotion Confidence Heatmap")
 
94
  heatmap_data.index = [item["sentence"] for item in data]
95
  return heatmap_data
96
 
97
+ def plot_emotion_heatmap(heatmap_data):
98
  fig = plt.figure(figsize=(10, len(data) * 0.5 + 2))
99
  sns.heatmap(heatmap_data, annot=True, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
100
  plt.title("Emotion Confidence Heatmap")