SpencerCPurdy commited on
Commit
c4b577f
·
verified ·
1 Parent(s): ff01b81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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