vickeee465 commited on
Commit
93eecfb
·
1 Parent(s): 61c786b

plt padding

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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("Emotion Confidence Heatmap")
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