Spaces:
Sleeping
Sleeping
Update src/rag_engine/build_vector_db.py
Browse files
src/rag_engine/build_vector_db.py
CHANGED
|
@@ -28,7 +28,9 @@ class MindGuardVectorDB:
|
|
| 28 |
|
| 29 |
# Initialize the Embedding Engine (all-MiniLM-L6-v2)
|
| 30 |
# This is a small, lightning-fast Hugging Face model that turns sentences into math vectors
|
| 31 |
-
self.embedding_fn = embedding_functions.
|
|
|
|
|
|
|
| 32 |
|
| 33 |
# Create or load the 'clinical_guidelines' collection
|
| 34 |
self.collection = self.chroma_client.get_or_create_collection(
|
|
|
|
| 28 |
|
| 29 |
# Initialize the Embedding Engine (all-MiniLM-L6-v2)
|
| 30 |
# This is a small, lightning-fast Hugging Face model that turns sentences into math vectors
|
| 31 |
+
self.embedding_fn = embedding_functions.SentenceTransformerEmbeddingFunction(
|
| 32 |
+
model_name="BAAI/bge-base-en-v1.5"
|
| 33 |
+
)
|
| 34 |
|
| 35 |
# Create or load the 'clinical_guidelines' collection
|
| 36 |
self.collection = self.chroma_client.get_or_create_collection(
|