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')
|