vickeee465 commited on
Commit
dd6393c
·
1 Parent(s): 09dd2f7

debugging print

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -76,6 +76,7 @@ def get_most_probable_label(probs):
76
  def prepare_heatmap_data(data):
77
  heatmap_data = pd.DataFrame(0, index=range(len(data)), columns=id2label.values())
78
  for idx, item in enumerate(data):
 
79
  for idy, confidence in enumerate(item["emotions"]):
80
  emotion = id2label[idy]
81
  heatmap_data.at[idx, emotion] = confidence
 
76
  def prepare_heatmap_data(data):
77
  heatmap_data = pd.DataFrame(0, index=range(len(data)), columns=id2label.values())
78
  for idx, item in enumerate(data):
79
+ print(item["emotions"]) # DEBUG
80
  for idy, confidence in enumerate(item["emotions"]):
81
  emotion = id2label[idy]
82
  heatmap_data.at[idx, emotion] = confidence