Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -8,7 +8,7 @@ from sklearn.feature_extraction.text import TfidfVectorizer
|
|
8 |
|
9 |
def main():
|
10 |
#####load dataset
|
11 |
-
dataset_1 = load_dataset("rotten_tomatoes")
|
12 |
dataset_2 = load_dataset('sst2')
|
13 |
dataset_2 = dataset_2.rename_column('sentence', 'text')
|
14 |
dataset_3 = load_dataset('imdb')
|
|
|
8 |
|
9 |
def main():
|
10 |
#####load dataset
|
11 |
+
dataset_1 = load_dataset("rotten_tomatoes", download_mode="force_redownload")
|
12 |
dataset_2 = load_dataset('sst2')
|
13 |
dataset_2 = dataset_2.rename_column('sentence', 'text')
|
14 |
dataset_3 = load_dataset('imdb')
|