Update src/models/summarization.py

src/models/summarization.py

@@ -21,8 +21,6 @@ class Summarizer:
             # Load the tokenizer
             self.tokenizer = BartTokenizer.from_pretrained("facebook/bart-base")
             
-            # Move model to appropriate device (GPU if available)
-            self.model.to(torch.device('cuda' if torch.cuda.is_available() else 'cpu'))
             return self.model
         except Exception as e:
             st.error(f"Error loading fine-tuned summarization model: {str(e)}")