Hugging Face's logo

Spaces:
thankrandomness
/
roberta-spam-detection-sample
Runtime error

App Files Community
thankrandomness commited on
Commit
70e635a
·
1 Parent(s): 97b93a8

load dataset

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from datasets import load_dataset
5
  dataset = load_dataset('thankrandomness/spam-detection-sample')
6
 
7
  # Convert to a pandas DataFrame
8
- df = pd.DataFrame(dataset['train'])
9
 
10
  # Display the first few rows of the dataframe to verify
11
  print(df.head())
 
5
  dataset = load_dataset('thankrandomness/spam-detection-sample')
6
 
7
  # Convert to a pandas DataFrame
8
+ df = pd.DataFrame(dataset)
9
 
10
  # Display the first few rows of the dataframe to verify
11
  print(df.head())