Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
app.py
CHANGED
@@ -71,7 +71,11 @@ llm = CustomInferenceClient(model_name=model_name, hf_token=hf_token, kwargs=kwa
|
|
71 |
|
72 |
# 임베딩 객체 생성
|
73 |
from langchain.embeddings import HuggingFaceInstructEmbeddings
|
74 |
-
embeddings = HuggingFaceInstructEmbeddings(
|
|
|
|
|
|
|
|
|
75 |
|
76 |
# 벡터DB 로드
|
77 |
path_work ='.'
|
|
|
71 |
|
72 |
# 임베딩 객체 생성
|
73 |
from langchain.embeddings import HuggingFaceInstructEmbeddings
|
74 |
+
embeddings = HuggingFaceInstructEmbeddings(
|
75 |
+
model_name="sentence-transformers/all-MiniLM-L6-v2",
|
76 |
+
cache_folder="./sentence-transformers/all-MiniLM-L6-v2",
|
77 |
+
model_kwargs={"device": "auto"}
|
78 |
+
)
|
79 |
|
80 |
# 벡터DB 로드
|
81 |
path_work ='.'
|