vives commited on
Commit
a8bcdc3
1 Parent(s): 09ac8a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -114,7 +114,7 @@ if text:
114
  st.json(sim_dict_finbert)
115
  else:
116
  idxs = extract_idxs(sim_dict, kp_dict)
117
- idxs_finbert = extract_idxs(sim_dict, kp_dict_finbert)
118
  distances_candidates = cosine_kp[np.ix_(idxs, idxs)]
119
  distances_candidates_finbert = cosine_finbert_kp[np.ix_(idxs_finbert, idxs_finbert)]
120
  #first do distilbert
 
114
  st.json(sim_dict_finbert)
115
  else:
116
  idxs = extract_idxs(sim_dict, kp_dict)
117
+ idxs_finbert = extract_idxs(sim_dict_finbert, kp_dict_finbert)
118
  distances_candidates = cosine_kp[np.ix_(idxs, idxs)]
119
  distances_candidates_finbert = cosine_finbert_kp[np.ix_(idxs_finbert, idxs_finbert)]
120
  #first do distilbert