Update src/summarizer.py

src/summarizer.py

@@ -1,4 +1,4 @@
-from transformers import AutoTokenizer, AutoModelForSeq2SeqSummarization
+from transformers import AutoTokenizer, AutoModelForSeq2SeqGeneration
 import torch
 import logging
 
@@ -16,7 +16,7 @@ class TextSummarizer:
             
             # Initialize tokenizer and model separately
             self.tokenizer = AutoTokenizer.from_pretrained(model_name)
-            self.model = AutoModelForSeq2SeqSummarization.from_pretrained(model_name)
+            self.model = AutoModelForSeq2SeqGeneration.from_pretrained(model_name)
             
             # Move model to CPU and eval mode
             self.model = self.model.to(self.device)