Spaces:
Sleeping
Sleeping
Update modules/dataset.py
Browse files- modules/dataset.py +1 -1
modules/dataset.py
CHANGED
@@ -6,7 +6,7 @@ emotions = dataset.info.features['labels'].feature.names
|
|
6 |
|
7 |
def query_emotion(start, end):
|
8 |
rows = dataset[start:end]
|
9 |
-
texts, labels = [rows[k] for k in rows.keys()]
|
10 |
|
11 |
observations = []
|
12 |
|
|
|
6 |
|
7 |
def query_emotion(start, end):
|
8 |
rows = dataset[start:end]
|
9 |
+
texts, labels, id = [rows[k] for k in rows.keys()]
|
10 |
|
11 |
observations = []
|
12 |
|