Update app.py

app.py

@@ -119,10 +119,10 @@ if text:
   with torch.no_grad():
     outputs = model(**new_tokens)
     outputs_finbert = model_finbert(**new_tokens_finbert)
-    outputs_sapbert = model_sapbert(**new_tokens_sapbert, pooler=False)
+    outputs_sapbert = model_sapbert(**new_tokens_sapbert)
   sim_dict = calculate_top_k(outputs, new_tokens, text, kp_dict, exclude_text=exclude_text,exclude_words=exclude_words,k=k)
   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)
-  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)
+  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)
   if not diversify_box:
     with col1:
       st.write("distilbert-cvent")