Update app.py

app.py

@@ -15,7 +15,9 @@ k = st.number_input("Top k nearest key-phrases",1,10)
 
 with open("kp_dict_merged.pickle",'rb') as handle:
   kp_dict = pickle.load(handle)
-
+for key in kp_dict.keys():
+  kp_dict[key] = kp_dict[key].detach().numpy()
+  
 def calculate_top_k(out, tokens,text,exclude_text=False,exclude_words=False, k=5):
   sim_dict = {}
   pools = pool_embeddings(out, tokens).detach().numpy()