cnp-ai commited on
Commit
c193aff
·
verified ·
1 Parent(s): ef06455

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ print(kpi_pool._collection.count())
30
 
31
  model_huggingface2 = HuggingFaceEmbeddings(model_name='jhgan/ko-sbert-sts')
32
  persist_directory_jhgan1 = './chroma_kpi_250707_SBERTjhgan'
33
- kpi_pool2 = Chroma(persist_directory=persist_directory_jhgan1, embedding_function=model_huggingface2, )
34
  print(kpi_pool2._collection.count())
35
 
36
 
 
30
 
31
  model_huggingface2 = HuggingFaceEmbeddings(model_name='jhgan/ko-sbert-sts')
32
  persist_directory_jhgan1 = './chroma_kpi_250707_SBERTjhgan'
33
+ kpi_pool2 = Chroma(persist_directory=persist_directory_jhgan1, embedding_function=model_huggingface2, collection_metadata={"hnsw:space": "cosine"})
34
  print(kpi_pool2._collection.count())
35
 
36