lunadebruyne commited on
Commit
7663872
·
1 Parent(s): e961250

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,12 +30,12 @@ def frequencies(preds):
30
  preds_dict = {"neutral": 0, "anger": 0, "fear": 0, "joy": 0, "love": 0, "sadness": 0}
31
  for pred in preds:
32
  preds_dict[pred] = preds_dict[pred] + 1
33
- labels = list(preds_dict.keys())
34
- x = list(preds_dict.values())
35
 
36
- fig, ax = plt.subplots(figsize=(6, 6))
37
- ax.pie(x, labels=labels, autopct='%.1f%%')
38
- ax.set_title('Emotion frequencies')
39
  return plt
40
 
41
  def inference_dataset(file_object, option_list):
 
30
  preds_dict = {"neutral": 0, "anger": 0, "fear": 0, "joy": 0, "love": 0, "sadness": 0}
31
  for pred in preds:
32
  preds_dict[pred] = preds_dict[pred] + 1
33
+ bars = list(preds_dict.keys())
34
+ height = list(preds_dict.values())
35
 
36
+ x_pos = np.arange(len(bars))
37
+ plt.bar(x_pos, height, color=['lightgrey', 'firebrick', 'rebeccapurple', 'orange', 'palevioletred', 'cornflowerblue'])
38
+ plt.xticks(x_pos, bars)
39
  return plt
40
 
41
  def inference_dataset(file_object, option_list):