truglpk3 commited on
Commit
59e6b8b
·
verified ·
1 Parent(s): c446249

Update chatbot/models/embeddings.py

Browse files
Files changed (1) hide show
  1. chatbot/models/embeddings.py +13 -12
chatbot/models/embeddings.py CHANGED
@@ -1,12 +1,13 @@
1
- from langchain_community.embeddings import HuggingFaceEmbeddings
2
-
3
- embeddings_model = None
4
-
5
- def load_model():
6
- global embeddings_model
7
- print("🚀 Bắt đầu load model Embedding...")
8
- embeddings_model = HuggingFaceEmbeddings(
9
- model_name="Alibaba-NLP/gte-multilingual-base",
10
- model_kwargs={"trust_remote_code": True}
11
- )
12
- print("✅ Embedding model loaded!")
 
 
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"]