cnp-ai commited on
Commit
0feee88
·
verified ·
1 Parent(s): b94bad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,8 +8,9 @@ import gradio as gr
8
  import numpy as np
9
 
10
  # Chroma DB 로드
11
- model_huggingface = HuggingFaceEmbeddings(model_name='BAAI/bge-m3')
12
- persist_directory1 = './chroma_kpi_250416'
 
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