puzan789 commited on
Commit
425e09d
·
1 Parent(s): cbeea95
Files changed (1) hide show
  1. src/embedding/embedding.py +17 -17
src/embedding/embedding.py CHANGED
@@ -1,27 +1,27 @@
1
- from langchain_huggingface import HuggingFaceEmbeddings
2
-
3
-
4
- embedding_instance = HuggingFaceEmbeddings(
5
- model_name="all-MiniLM-L6-v2",
6
- model_kwargs={"device": "cpu"},
7
- )
8
-
9
- def all_minilm_l6_v2():
10
- """
11
- Return the embedding instance.
12
- """
13
- return embedding_instance
14
-
15
-
16
  # from langchain_huggingface import HuggingFaceEmbeddings
17
 
 
18
  # embedding_instance = HuggingFaceEmbeddings(
19
- # model_name="sentence-transformers/all-MiniLM-L6-v2",
20
  # model_kwargs={"device": "cpu"},
21
  # )
22
 
23
  # def all_minilm_l6_v2():
24
  # """
25
- # Return the embedding instance.
26
  # """
27
  # return embedding_instance
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # from langchain_huggingface import HuggingFaceEmbeddings
2
 
3
+
4
  # embedding_instance = HuggingFaceEmbeddings(
5
+ # model_name="all-MiniLM-L6-v2",
6
  # model_kwargs={"device": "cpu"},
7
  # )
8
 
9
  # def all_minilm_l6_v2():
10
  # """
11
+ # Return the embedding instance.
12
  # """
13
  # return embedding_instance
14
+
15
+
16
+ from langchain_huggingface import HuggingFaceEmbeddings
17
+
18
+ embedding_instance = HuggingFaceEmbeddings(
19
+ model_name="sentence-transformers/all-MiniLM-L6-v2",
20
+ model_kwargs={"device": "cpu"},
21
+ )
22
+
23
+ def all_minilm_l6_v2():
24
+ """
25
+ Return the embedding instance.
26
+ """
27
+ return embedding_instance