pritamdeka commited on
Commit
ce29488
Β·
1 Parent(s): ab9ffdb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -115,7 +115,7 @@ def keyphrase_generator(article_link, model_1, model_2, max_num_keywords, model_
115
  elif(len_embeddings == 768):
116
  sim_mat[i][j] = cosine_similarity(corpus_embeddings[i].reshape(1,768), corpus_embeddings[j].reshape(1,768))[0,0]
117
  nx_graph = nx.from_numpy_array(sim_mat)
118
- scores = nx.pagerank(nx_graph)
119
  sentences=((scores[i],s) for i,s in enumerate(corpus))
120
  for elem in sentences:
121
  element.append(elem[0])
 
115
  elif(len_embeddings == 768):
116
  sim_mat[i][j] = cosine_similarity(corpus_embeddings[i].reshape(1,768), corpus_embeddings[j].reshape(1,768))[0,0]
117
  nx_graph = nx.from_numpy_array(sim_mat)
118
+ scores = nx.pagerank(nx_graph, max_iter = 1500)
119
  sentences=((scores[i],s) for i,s in enumerate(corpus))
120
  for elem in sentences:
121
  element.append(elem[0])