Spaces:
Running
Running
michaelkri commited on
Commit ·
ac34fec
1
Parent(s): 037df93
Changed default parameter
Browse files- app/summarizer.py +3 -3
app/summarizer.py
CHANGED
|
@@ -74,7 +74,7 @@ class Summarizer:
|
|
| 74 |
return ranked_sentences
|
| 75 |
|
| 76 |
|
| 77 |
-
def summarize_clusters(self, clusters, top_cluster_count=
|
| 78 |
# sort clusters by their length (descending) to find the most important topics
|
| 79 |
clusters = sorted(clusters, key=len, reverse=True)
|
| 80 |
|
|
@@ -86,7 +86,7 @@ class Summarizer:
|
|
| 86 |
for i, cluster in enumerate(clusters):
|
| 87 |
logging.debug(f'Extracting from cluster {i + 1}...')
|
| 88 |
top_sentences = Summarizer.rank_cluster_sentences(cluster)
|
| 89 |
-
content = '\n'.join(top_sentences[:
|
| 90 |
key_sentences.append(content)
|
| 91 |
|
| 92 |
combined = ' '.join(key_sentences)
|
|
@@ -101,7 +101,7 @@ class Summarizer:
|
|
| 101 |
return summary
|
| 102 |
|
| 103 |
|
| 104 |
-
def multisource_summary(self, articles, min_cluster_size=
|
| 105 |
'''
|
| 106 |
Create a single summary from multiple articles
|
| 107 |
'''
|
|
|
|
| 74 |
return ranked_sentences
|
| 75 |
|
| 76 |
|
| 77 |
+
def summarize_clusters(self, clusters, top_cluster_count=10, top_k_sentences=10):
|
| 78 |
# sort clusters by their length (descending) to find the most important topics
|
| 79 |
clusters = sorted(clusters, key=len, reverse=True)
|
| 80 |
|
|
|
|
| 86 |
for i, cluster in enumerate(clusters):
|
| 87 |
logging.debug(f'Extracting from cluster {i + 1}...')
|
| 88 |
top_sentences = Summarizer.rank_cluster_sentences(cluster)
|
| 89 |
+
content = '\n'.join(top_sentences[:top_k_sentences])
|
| 90 |
key_sentences.append(content)
|
| 91 |
|
| 92 |
combined = ' '.join(key_sentences)
|
|
|
|
| 101 |
return summary
|
| 102 |
|
| 103 |
|
| 104 |
+
def multisource_summary(self, articles, min_cluster_size=2):
|
| 105 |
'''
|
| 106 |
Create a single summary from multiple articles
|
| 107 |
'''
|