Update knowledgeassistant/components/summarization.py

knowledgeassistant/components/summarization.py

@@ -19,13 +19,9 @@ class DataSummarization:
         try:
             pipe = pipeline("summarization", model="/app/models/bart-large-cnn")
             logging.info("Summarization Pipeline Successfully Setup")
-
             text = read_txt_file(input_text_path)
-            print(f"Input Text: {text[:500]}")
             summary = pipe(text, min_length = min_length, do_sample = False)
-            print(f"Summary Output: {summary}")
             logging.info("Text successfully summarized")
-            
             write_txt_file(self.data_summarization_config.summarized_text_file_path, summary[0].get("summary_text"))
             logging.info("Successfully wrote summarized text")
         except Exception as e: