Spaces:
Running
Running
vkovacs
commited on
Commit
·
7183843
1
Parent(s):
9a986f5
making it slightly taller
Browse files
app.py
CHANGED
@@ -87,7 +87,7 @@ def prepare_heatmap_data(data):
|
|
87 |
return heatmap_data
|
88 |
|
89 |
def plot_emotion_heatmap(heatmap_data):
|
90 |
-
fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 2,
|
91 |
sns.heatmap(heatmap_data, annot=False, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
|
92 |
plt.title("Sentence-level emotion confidences")
|
93 |
plt.xlabel("Sentences")
|
|
|
87 |
return heatmap_data
|
88 |
|
89 |
def plot_emotion_heatmap(heatmap_data):
|
90 |
+
fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 2, 5))
|
91 |
sns.heatmap(heatmap_data, annot=False, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
|
92 |
plt.title("Sentence-level emotion confidences")
|
93 |
plt.xlabel("Sentences")
|