Spaces:
Running
Running
Update chatbot/models/embeddings.py
Browse files- chatbot/models/embeddings.py +13 -12
chatbot/models/embeddings.py
CHANGED
|
@@ -1,12 +1,13 @@
|
|
| 1 |
-
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
|
|
|
|
|
| 1 |
+
from langchain_community.embeddings import HuggingFaceEmbeddings
|
| 2 |
+
|
| 3 |
+
def get_embeddings():
|
| 4 |
+
model_name = "Alibaba-NLP/gte-multilingual-base"
|
| 5 |
+
embeddings = HuggingFaceEmbeddings(
|
| 6 |
+
model_name=model_name,
|
| 7 |
+
model_kwargs={"trust_remote_code": True}
|
| 8 |
+
)
|
| 9 |
+
return embeddings
|
| 10 |
+
|
| 11 |
+
embeddings = get_embeddings()
|
| 12 |
+
|
| 13 |
+
__all__ = ["embeddings", "get_embeddings"]
|