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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def calculate_top_k(out, tokens,text,kp_dict,exclude_text=False,exclude_words=Fa
60
  if pooler:
61
  pools = pool_embeddings(out, tokens).detach().numpy()
62
  else:
63
- pools = out["pooler_outputs"].detach().numpy()
64
  for key in kp_dict.keys():
65
  if key == text:
66
  continue
 
60
  if pooler:
61
  pools = pool_embeddings(out, tokens).detach().numpy()
62
  else:
63
+ pools = out["pooler_output"].detach().numpy()
64
  for key in kp_dict.keys():
65
  if key == text:
66
  continue