Update app.py
Browse files
app.py
CHANGED
|
@@ -218,8 +218,8 @@ class NewsGroupsDataLoader:
|
|
| 218 |
test_data = dataset['test']
|
| 219 |
|
| 220 |
# Combine for proper splitting
|
| 221 |
-
all_texts = train_data['text'] + test_data['text']
|
| 222 |
-
all_labels = train_data['label'] + test_data['label']
|
| 223 |
|
| 224 |
# Get category names from dataset features
|
| 225 |
self.categories = train_data.features['label'].names
|
|
|
|
| 218 |
test_data = dataset['test']
|
| 219 |
|
| 220 |
# Combine for proper splitting
|
| 221 |
+
all_texts = list(train_data['text']) + list(test_data['text'])
|
| 222 |
+
all_labels = list(train_data['label']) + list(test_data['label'])
|
| 223 |
|
| 224 |
# Get category names from dataset features
|
| 225 |
self.categories = train_data.features['label'].names
|