Spaces:
Sleeping
Sleeping
Update modules/extractive.py
Browse files- modules/extractive.py +2 -6
modules/extractive.py
CHANGED
@@ -40,12 +40,8 @@ class CombinedSummarizer:
|
|
40 |
|
41 |
|
42 |
class BERTSummarizer:
|
43 |
-
@staticmethod
|
44 |
def __init__(self):
|
45 |
self.model = Summarizer()
|
46 |
|
47 |
-
def summarize(self,
|
48 |
-
text =
|
49 |
-
|
50 |
-
summary = self.model(text, num_sentences=num_sentences)
|
51 |
-
return summary
|
|
|
40 |
|
41 |
|
42 |
class BERTSummarizer:
|
|
|
43 |
def __init__(self):
|
44 |
self.model = Summarizer()
|
45 |
|
46 |
+
def summarize(self, text, num_sentences):
|
47 |
+
return self.model(text, num_sentences=num_sentences)
|
|
|
|
|
|