Spaces:
Running
Running
Commit
•
f4920d0
1
Parent(s):
ed9ae2b
Fix error (#2)
Browse files- Fix error (5997a7d618cd473f320a6844c640a10c1b9d07e9)
Co-authored-by: Christopher Akiki <[email protected]>
- dataset.py +1 -1
dataset.py
CHANGED
@@ -2,7 +2,7 @@ from datasets import load_dataset
|
|
2 |
|
3 |
dataset = load_dataset("go_emotions", split="train")
|
4 |
|
5 |
-
emotions = dataset.info.features[
|
6 |
|
7 |
def query_emotion(start, end):
|
8 |
rows = dataset[start:end]
|
|
|
2 |
|
3 |
dataset = load_dataset("go_emotions", split="train")
|
4 |
|
5 |
+
emotions = dataset.info.features['labels'].feature.names
|
6 |
|
7 |
def query_emotion(start, end):
|
8 |
rows = dataset[start:end]
|