Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ for item in emails.message:
|
|
| 31 |
content_text.append(cleaned_message)
|
| 32 |
|
| 33 |
# Taking a sample of the dataset
|
| 34 |
-
train, test = train_test_split(content_text, train_size = 0.
|
| 35 |
|
| 36 |
# Setting up ids for ChromaDB collections
|
| 37 |
ids = []
|
|
|
|
| 31 |
content_text.append(cleaned_message)
|
| 32 |
|
| 33 |
# Taking a sample of the dataset
|
| 34 |
+
train, test = train_test_split(content_text, train_size = 0.001) # Dataset is too large to complete embedding step
|
| 35 |
|
| 36 |
# Setting up ids for ChromaDB collections
|
| 37 |
ids = []
|