movie_title_gpt2

Runtime error

radames commited on Oct 9, 2023

Commit

9e0e6dd

•

1 Parent(s): aaeea6a

Update modules/dataset.py

Files changed (1) hide show

modules/dataset.py CHANGED Viewed

@@ -6,7 +6,7 @@ emotions = dataset.info.features['labels'].feature.names
 def query_emotion(start, end):
     rows = dataset[start:end]
-    texts, labels = [rows[k] for k in rows.keys()]
     observations = []

 def query_emotion(start, end):
     rows = dataset[start:end]
+    texts, labels, id = [rows[k] for k in rows.keys()]
     observations = []