Bhanuprasadchouki commited on
Commit
d17c739
·
verified ·
1 Parent(s): c2b7f17

Update rag2.py

Browse files
Files changed (1) hide show
  1. rag2.py +1 -1
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-transformers/all-mpnet-base-v2"
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)