Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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")
|