Update app.py
Browse files
app.py
CHANGED
@@ -2,13 +2,15 @@ import nltk
|
|
2 |
nltk.download('punkt')
|
3 |
from nltk.tokenize import sent_tokenize
|
4 |
import chromadb
|
|
|
|
|
5 |
|
6 |
# Load the email dataset
|
7 |
# emails = pd.read_csv("/content/drive/MyDrive/Clean/cleaned_data.csv")
|
8 |
|
9 |
|
10 |
client = chromadb.Client()
|
11 |
-
client = chromadb.PersistentClient(path="
|
12 |
|
13 |
# Load the ChromaDB collection
|
14 |
collection = client.get_collection("enron_emails")
|
|
|
2 |
nltk.download('punkt')
|
3 |
from nltk.tokenize import sent_tokenize
|
4 |
import chromadb
|
5 |
+
from chromadb.utils import embedding_functions
|
6 |
+
|
7 |
|
8 |
# Load the email dataset
|
9 |
# emails = pd.read_csv("/content/drive/MyDrive/Clean/cleaned_data.csv")
|
10 |
|
11 |
|
12 |
client = chromadb.Client()
|
13 |
+
client = chromadb.PersistentClient(path="blob/main/chroma.sqlite3")
|
14 |
|
15 |
# Load the ChromaDB collection
|
16 |
collection = client.get_collection("enron_emails")
|