AndreMitri commited on
Commit
2fc85f6
1 Parent(s): 8589dd2

Changed APP dataset from IMDB to ROTTEN TOMATOS

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
  from transformers import AutoTokenizer, pipeline
6
  from wordnet import wordnet_pipeline
7
 
8
- dataset = load_dataset('danielcd99/imdb')
9
 
10
  dataframes = {}
11
  for split in dataset.keys():
 
5
  from transformers import AutoTokenizer, pipeline
6
  from wordnet import wordnet_pipeline
7
 
8
+ dataset = load_dataset('AndreMitri/rotten_tomatos')
9
 
10
  dataframes = {}
11
  for split in dataset.keys():