cnp-ai commited on
Commit
69f55cc
·
verified ·
1 Parent(s): be02948

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -109,10 +109,13 @@ def filter_semantically_similar_texts_by_embedding(df, mode, embedding_field='em
109
 
110
  def search_kpi(kpi_query, kpi_count, mode):
111
  if mode == "BGE":
 
112
  results = kpi_pool_ori.similarity_search_with_relevance_scores(kpi_query, k=50)
113
  elif mode == "SBERT-snunlp":
 
114
  results = kpi_pool.similarity_search_with_relevance_scores(kpi_query, k=50)
115
  else:
 
116
  results = kpi_pool2.similarity_search_with_relevance_scores(kpi_query, k=50)
117
 
118
 
@@ -142,7 +145,7 @@ def search_kpi_all_models(kpi_query, kpi_count):
142
  _, _, df_bge, _, _, _ = search_kpi(kpi_query, kpi_count, "BGE")
143
  _, _, df_sn, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-snunlp")
144
  _, _, df_jh, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-jhgan")
145
-
146
  return df_bge, df_sn, df_jh
147
 
148
 
 
109
 
110
  def search_kpi(kpi_query, kpi_count, mode):
111
  if mode == "BGE":
112
+ print("BGE 검색 시작")
113
  results = kpi_pool_ori.similarity_search_with_relevance_scores(kpi_query, k=50)
114
  elif mode == "SBERT-snunlp":
115
+ print("SBERT-snunlp 검색 시작")
116
  results = kpi_pool.similarity_search_with_relevance_scores(kpi_query, k=50)
117
  else:
118
+ print("SBERT-jhgan 검색 시작")
119
  results = kpi_pool2.similarity_search_with_relevance_scores(kpi_query, k=50)
120
 
121
 
 
145
  _, _, df_bge, _, _, _ = search_kpi(kpi_query, kpi_count, "BGE")
146
  _, _, df_sn, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-snunlp")
147
  _, _, df_jh, _, _, _ = search_kpi(kpi_query, kpi_count, "SBERT-jhgan")
148
+ print("함수 종료")
149
  return df_bge, df_sn, df_jh
150
 
151