Update embeddings.py
Browse files
climateqa/engine/embeddings.py
CHANGED
|
@@ -11,7 +11,7 @@ def get_embeddings_function(version = "v1.2"):
|
|
| 11 |
|
| 12 |
model_name = "BAAI/bge-base-en-v1.5"
|
| 13 |
encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
|
| 14 |
-
embeddings_function =
|
| 15 |
model_name=model_name,
|
| 16 |
encode_kwargs=encode_kwargs,
|
| 17 |
query_instruction="Represent this sentence for searching relevant passages: "
|
|
|
|
| 11 |
|
| 12 |
model_name = "BAAI/bge-base-en-v1.5"
|
| 13 |
encode_kwargs = {'normalize_embeddings': True} # set True to compute cosine similarity
|
| 14 |
+
embeddings_function = HuggingFaceEmbeddings(
|
| 15 |
model_name=model_name,
|
| 16 |
encode_kwargs=encode_kwargs,
|
| 17 |
query_instruction="Represent this sentence for searching relevant passages: "
|