vives commited on
Commit
56eac52
1 Parent(s): 06da3ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ if text:
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_kp_finbert[np.ix_(idxs_finbert, idxs_finbert)]
120
  #first do distilbert
121
  candidate = None
122
  min_sim = np.inf
 
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
121
  candidate = None
122
  min_sim = np.inf