Spaces:
Sleeping
Sleeping
Update rag2.py
Browse files
rag2.py
CHANGED
@@ -7,7 +7,7 @@ from langchain_huggingface import HuggingFaceEmbeddings
|
|
7 |
# Configuration
|
8 |
GOOGLE_API_KEY = "AIzaSyB-7cKMdUpA5kTccpNxd72IT5CjeSgSmkc" # Replace with your API key
|
9 |
CHROMA_DB_DIR = "./chroma_db_" # Directory for ChromaDB
|
10 |
-
MODEL_NAME = "sentence-
|
11 |
|
12 |
# Initialize HuggingFace Embeddings
|
13 |
embeddings_model = HuggingFaceEmbeddings(model_name=MODEL_NAME)
|
|
|
7 |
# Configuration
|
8 |
GOOGLE_API_KEY = "AIzaSyB-7cKMdUpA5kTccpNxd72IT5CjeSgSmkc" # Replace with your API key
|
9 |
CHROMA_DB_DIR = "./chroma_db_" # Directory for ChromaDB
|
10 |
+
MODEL_NAME = "flax-sentence-embeddings/all_datasets_v4_MiniLM-L6"
|
11 |
|
12 |
# Initialize HuggingFace Embeddings
|
13 |
embeddings_model = HuggingFaceEmbeddings(model_name=MODEL_NAME)
|