vives commited on
Commit
8767f7f
·
1 Parent(s): e9c9402

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -119,10 +119,10 @@ if text:
119
  with torch.no_grad():
120
  outputs = model(**new_tokens)
121
  outputs_finbert = model_finbert(**new_tokens_finbert)
122
- outputs_sapbert = model_sapbert(**new_tokens_sapbert, pooler=False)
123
  sim_dict = calculate_top_k(outputs, new_tokens, text, kp_dict, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
124
  sim_dict_finbert = calculate_top_k(outputs_finbert, new_tokens_finbert, text, kp_dict_finbert, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
125
- sim_dict_sapbert = calculate_top_k(outputs_sapbert, new_tokens_sapbert, text, kp_dict_sapbert, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
126
  if not diversify_box:
127
  with col1:
128
  st.write("distilbert-cvent")
 
119
  with torch.no_grad():
120
  outputs = model(**new_tokens)
121
  outputs_finbert = model_finbert(**new_tokens_finbert)
122
+ outputs_sapbert = model_sapbert(**new_tokens_sapbert)
123
  sim_dict = calculate_top_k(outputs, new_tokens, text, kp_dict, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
124
  sim_dict_finbert = calculate_top_k(outputs_finbert, new_tokens_finbert, text, kp_dict_finbert, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
125
+ sim_dict_sapbert = calculate_top_k(outputs_sapbert, new_tokens_sapbert, text, kp_dict_sapbert, exclude_text=exclude_text,exclude_words=exclude_words,k=k, pooler=False)
126
  if not diversify_box:
127
  with col1:
128
  st.write("distilbert-cvent")