Spaces:
Running
Running
vickeee465
commited on
Commit
·
dd6393c
1
Parent(s):
09dd2f7
debugging print
Browse files
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
|