Update db.py

db.py

@@ -35,7 +35,7 @@ def split_data(data, chunk_size, chunk_overlap):
     return splitter.split_documents(data)
 
 
-def generate_embeddings():
+def generate_embeddings(model_path):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model_kwargs = {'device': device} 
 
@@ -57,7 +57,7 @@ def get_db(
     
     splits = split_data(data, chunk_size, chunk_overlap)
 
-    embedding = generate_embeddings()
+    embedding = generate_embeddings(model_path)
     
     return Chroma.from_documents(
         documents=splits,