Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -119,7 +119,7 @@ def get_text_chunks(documents):
|
|
| 119 |
# ํ
์คํธ ์ฒญํฌ๋ค๋ก๋ถํฐ ๋ฒกํฐ ์คํ ์ด๋ฅผ ์์ฑํ๋ ํจ์์
๋๋ค.
|
| 120 |
def get_vectorstore(text_chunks):
|
| 121 |
# ์ํ๋ ์๋ฒ ๋ฉ ๋ชจ๋ธ์ ๋ก๋ํฉ๋๋ค.
|
| 122 |
-
embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-
|
| 123 |
model_kwargs={'device': 'cpu'}) # ์๋ฒ ๋ฉ ๋ชจ๋ธ์ ์ค์ ํฉ๋๋ค.
|
| 124 |
vectorstore = FAISS.from_documents(text_chunks, embeddings) # FAISS ๋ฒกํฐ ์คํ ์ด๋ฅผ ์์ฑํฉ๋๋ค.
|
| 125 |
return vectorstore # ์์ฑ๋ ๋ฒกํฐ ์คํ ์ด๋ฅผ ๋ฐํํฉ๋๋ค.
|
|
|
|
| 119 |
# ํ
์คํธ ์ฒญํฌ๋ค๋ก๋ถํฐ ๋ฒกํฐ ์คํ ์ด๋ฅผ ์์ฑํ๋ ํจ์์
๋๋ค.
|
| 120 |
def get_vectorstore(text_chunks):
|
| 121 |
# ์ํ๋ ์๋ฒ ๋ฉ ๋ชจ๋ธ์ ๋ก๋ํฉ๋๋ค.
|
| 122 |
+
embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2',
|
| 123 |
model_kwargs={'device': 'cpu'}) # ์๋ฒ ๋ฉ ๋ชจ๋ธ์ ์ค์ ํฉ๋๋ค.
|
| 124 |
vectorstore = FAISS.from_documents(text_chunks, embeddings) # FAISS ๋ฒกํฐ ์คํ ์ด๋ฅผ ์์ฑํฉ๋๋ค.
|
| 125 |
return vectorstore # ์์ฑ๋ ๋ฒกํฐ ์คํ ์ด๋ฅผ ๋ฐํํฉ๋๋ค.
|