Spaces:
Sleeping
Sleeping
vickeee465
commited on
Commit
·
93eecfb
1
Parent(s):
61c786b
plt padding
Browse files
app.py
CHANGED
|
@@ -97,9 +97,10 @@ def prepare_heatmap_data(data):
|
|
| 97 |
def plot_emotion_heatmap(heatmap_data):
|
| 98 |
fig = plt.figure(figsize=(10, len(heatmap_data) * 0.5 + 2))
|
| 99 |
sns.heatmap(heatmap_data, annot=True, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
|
| 100 |
-
plt.title("
|
| 101 |
plt.xlabel("Emotions")
|
| 102 |
plt.ylabel("Sentences")
|
|
|
|
| 103 |
plt.tight_layout()
|
| 104 |
return fig
|
| 105 |
|
|
|
|
| 97 |
def plot_emotion_heatmap(heatmap_data):
|
| 98 |
fig = plt.figure(figsize=(10, len(heatmap_data) * 0.5 + 2))
|
| 99 |
sns.heatmap(heatmap_data, annot=True, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
|
| 100 |
+
plt.title("Sentence-level emotion confidences")
|
| 101 |
plt.xlabel("Emotions")
|
| 102 |
plt.ylabel("Sentences")
|
| 103 |
+
plt.subplots_adjust(left=0.2, right=0.95, top=0.9, bottom=0.2)
|
| 104 |
plt.tight_layout()
|
| 105 |
return fig
|
| 106 |
|