Update app.py
Browse files
app.py
CHANGED
@@ -37,8 +37,8 @@ def load_embedding_model():
|
|
37 |
return SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
38 |
|
39 |
@st.cache_resource
|
40 |
-
def load_vectorstore(
|
41 |
-
embedding_function = CustomEmbedding(
|
42 |
return Chroma(persist_directory="./chroma_batch_vectors", embedding_function=embedding_function)
|
43 |
|
44 |
# ์ง๋ฌธ์ ๋ํ ์๋ต ์์ฑ ํจ์
|
@@ -72,3 +72,4 @@ if user_input:
|
|
72 |
del response
|
73 |
gc.collect()
|
74 |
|
|
|
|
37 |
return SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
38 |
|
39 |
@st.cache_resource
|
40 |
+
def load_vectorstore(_embedding_model): # _embedding_model๋ก ์ด๋ฆ ๋ณ๊ฒฝ
|
41 |
+
embedding_function = CustomEmbedding(_embedding_model)
|
42 |
return Chroma(persist_directory="./chroma_batch_vectors", embedding_function=embedding_function)
|
43 |
|
44 |
# ์ง๋ฌธ์ ๋ํ ์๋ต ์์ฑ ํจ์
|
|
|
72 |
del response
|
73 |
gc.collect()
|
74 |
|
75 |
+
|