vkovacs commited on
Commit
6947b3f
·
1 Parent(s): 0c248a7

making it nicer

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, 5))
91
  sns.heatmap(heatmap_data, annot=False, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
92
  plt.xlabel("Sentences")
93
  plt.ylabel("Emotions")
 
87
  return heatmap_data
88
 
89
  def plot_emotion_heatmap(heatmap_data):
90
+ fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, 5))
91
  sns.heatmap(heatmap_data, annot=False, cmap="coolwarm", cbar=True, linewidths=0.5, linecolor='gray')
92
  plt.xlabel("Sentences")
93
  plt.ylabel("Emotions")