Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
5250894
1
Parent(s):
74fed9b
variable name
Browse files
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(
|
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")
|