Spaces:
Runtime error
Runtime error
Commit
Β·
ce29488
1
Parent(s):
ab9ffdb
Update app.py
Browse files
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])
|