Update modules/extractive.py

modules/extractive.py

@@ -40,8 +40,12 @@ class CombinedSummarizer:
 
 
 class BERTSummarizer:
+    @staticmethod
     def __init__(self):
         self.model = Summarizer()
 
-    def summarize(self, text, num_sentences):
-        return self.model(text, num_sentences=num_sentences)
+    def summarize(self, sentences, preprocessed_sentences, num_sentences):
+        text = " ".join(sentences)
+    
+        summary = self.model(text, num_sentences=num_sentences)
+        return summary