Update app.py

app.py

@@ -1,12 +1,18 @@
 import nltk
 nltk.download('punkt')
 from nltk.tokenize import sent_tokenize
+import chromadb
 
 # Load the email dataset
 # emails = pd.read_csv("/content/drive/MyDrive/Clean/cleaned_data.csv")
 
+
+client = chromadb.Client()
 client = chromadb.PersistentClient(path="./")
 
+# Load the ChromaDB collection
+collection = client.get_collection("enron_emails")
+
 # Create a ChromaDB client
 # client = chromadb.Client()
 # collection = client.create_collection("enron_emails")
@@ -27,10 +33,6 @@ model = AutoModelForSeq2SeqLM.from_pretrained("google-t5/t5-small")
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained("google-t5/t5-small")
 
-# Load the ChromaDB collection
-client = chromadb.Client()
-collection = client.get_collection("enron_emails")
-
 def query_collection(query_text):
     try:
         # Perform the query