Update knowledgeassistant/components/summarization.py

knowledgeassistant/components/summarization.py

@@ -17,8 +17,7 @@ class DataSummarization:
 
     def summarize(self, input_text_path: str, min_length: int):
         try:
-            device = "cuda" if torch.cuda.is_available() else "cpu"
-            pipe = pipeline("summarization", model="facebook/bart-large-cnn", device=0 if device == "cuda" else -1)
+            pipe = pipeline("summarization", model="facebook/bart-large-cnn")
             logging.info("Summarization Pipeline Successfully Setup")
 
             text = read_txt_file(input_text_path)