Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,9 @@ k = st.number_input("Top k nearest key-phrases",1,10)
|
|
15 |
|
16 |
with open("kp_dict_merged.pickle",'rb') as handle:
|
17 |
kp_dict = pickle.load(handle)
|
18 |
-
|
|
|
|
|
19 |
def calculate_top_k(out, tokens,text,exclude_text=False,exclude_words=False, k=5):
|
20 |
sim_dict = {}
|
21 |
pools = pool_embeddings(out, tokens).detach().numpy()
|
|
|
15 |
|
16 |
with open("kp_dict_merged.pickle",'rb') as handle:
|
17 |
kp_dict = pickle.load(handle)
|
18 |
+
for key in kp_dict.keys():
|
19 |
+
kp_dict[key] = kp_dict[key].detach().numpy()
|
20 |
+
|
21 |
def calculate_top_k(out, tokens,text,exclude_text=False,exclude_words=False, k=5):
|
22 |
sim_dict = {}
|
23 |
pools = pool_embeddings(out, tokens).detach().numpy()
|