Update app.py
Browse files
app.py
CHANGED
@@ -8,8 +8,9 @@ import gradio as gr
|
|
8 |
import numpy as np
|
9 |
|
10 |
# Chroma DB 로드
|
11 |
-
|
12 |
-
|
|
|
13 |
kpi_pool = Chroma(persist_directory=persist_directory1, embedding_function=model_huggingface)
|
14 |
print(kpi_pool._collection.count())
|
15 |
|
|
|
8 |
import numpy as np
|
9 |
|
10 |
# Chroma DB 로드
|
11 |
+
|
12 |
+
model_huggingface = HuggingFaceEmbeddings(model_name='snunlp/KR-SBERT-V40K-klueNLI-augSTS')
|
13 |
+
persist_directory1 = './chroma_kpi_250528_SBERT'
|
14 |
kpi_pool = Chroma(persist_directory=persist_directory1, embedding_function=model_huggingface)
|
15 |
print(kpi_pool._collection.count())
|
16 |
|