Overglitch commited on
Commit
b6e8059
·
verified ·
1 Parent(s): b6c116a

Update modules/extractive.py

Browse files
Files changed (1) hide show
  1. 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, sentences, preprocessed_sentences, num_sentences):
48
- text = " ".join(sentences)
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)