Spaces:
Running
Running
Update retrival.py
Browse files- retrival.py +2 -2
retrival.py
CHANGED
|
@@ -271,8 +271,8 @@ def save_to_chroma(chunks: list[Document], name: str):
|
|
| 271 |
|
| 272 |
try:
|
| 273 |
# Load the embedding model
|
| 274 |
-
|
| 275 |
-
embedding_function = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1")
|
| 276 |
# Create Chroma DB for documents using from_documents [NOTE: Some of the data is converted to string because int and float show null if added]
|
| 277 |
print("Creating document vector database...")
|
| 278 |
db = Chroma.from_documents(
|
|
|
|
| 271 |
|
| 272 |
try:
|
| 273 |
# Load the embedding model
|
| 274 |
+
embedding_function = HuggingFaceEmbeddings(model_name="all-MiniLM-L6-v2")
|
| 275 |
+
#embedding_function = HuggingFaceEmbeddings(model_name="mixedbread-ai/mxbai-embed-large-v1")
|
| 276 |
# Create Chroma DB for documents using from_documents [NOTE: Some of the data is converted to string because int and float show null if added]
|
| 277 |
print("Creating document vector database...")
|
| 278 |
db = Chroma.from_documents(
|