hewoo commited on
Commit
c4f01b6
ยท
verified ยท
1 Parent(s): 1dc17cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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(embedding_model):
41
- embedding_function = CustomEmbedding(embedding_model)
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
+