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