m

requirements.txt

@@ -1,5 +1,6 @@
 bs4
-streamlit==1.30.0
+streamlit
+lxml
 torch
 requests
 transformers

src/summarizer.py

@@ -11,6 +11,6 @@ class Summarizer:
 
     def summarize(self, text):
         input_ids = self.tokenizer(text, return_tensors="pt").input_ids.to(self.device)
-        outputs = self.model.generate(input_ids, max_length=20, min_length=20, length_penalty=2.0, num_beams=5, early_stopping=True)
+        outputs = self.model.generate(input_ids, max_length=20, min_length=20, length_penalty=2.0, num_beams=2, early_stopping=True)
 
         return self.tokenizer.decode(outputs[0], skip_special_tokens=True)