Fixed some bugs

DeevsDeevs · Oct 23, 2022 · 53aa2e8 · 53aa2e8
1 parent b1b51a4
commit 53aa2e8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/streamlit-app/utils.py b/streamlit-app/utils.py
@@ -152,7 +152,8 @@ def get_search_results(search_request: str, additional_info: str, data: pd.DataF
     if bm25 != None:
         ids, scores = bm25.predict(search_request)
         bm25_top10 = goods_df.iloc[ids].head(10) # type: ignore
-        bm25_top10['kpgz_sim'] = bm25_top10['Код КПГЗ'].apply(lambda x: calc_kpgz(x, kpgz_table))
+        # bm25_top10['kpgz_sim'] = bm25_top10['Код КПГЗ'].apply(lambda x: calc_kpgz(x, kpgz_table))
+        bm25_top10['kpgz_sim'] = 0 
 
     faiss.normalize_L2(xq)
     k = 100