Spaces:
Runtime error
Runtime error
Update initialize.py
Browse files- initialize.py +2 -2
initialize.py
CHANGED
@@ -13,9 +13,9 @@ OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY')
|
|
13 |
# Using Google GenAI Text Embeddings
|
14 |
# embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY) ###
|
15 |
|
16 |
-
|
17 |
|
18 |
-
embedding_model = OllamaEmbeddings(model="mxbai-embed-large")
|
19 |
|
20 |
# Create Embeddings for Searching the Splits
|
21 |
persist_directory = './chroma/'
|
|
|
13 |
# Using Google GenAI Text Embeddings
|
14 |
# embedding_model = GoogleGenerativeAIEmbeddings(model="models/embedding-001", task_type="retrieval_query", google_api_key=GEMINI_API_KEY) ###
|
15 |
|
16 |
+
embedding_model = OpenAIEmbeddings(api_key=OPENAI_API_KEY, model="text-embedding-3-small")
|
17 |
|
18 |
+
#embedding_model = OllamaEmbeddings(model="mxbai-embed-large")
|
19 |
|
20 |
# Create Embeddings for Searching the Splits
|
21 |
persist_directory = './chroma/'
|