Spaces:
Running
Running
Update knowledgeassistant/components/summarization.py
Browse files
knowledgeassistant/components/summarization.py
CHANGED
@@ -17,7 +17,7 @@ class DataSummarization:
|
|
17 |
|
18 |
def summarize(self, input_text_path: str, min_length: int):
|
19 |
try:
|
20 |
-
pipe = pipeline("summarization", model="
|
21 |
logging.info("Summarization Pipeline Successfully Setup")
|
22 |
|
23 |
text = read_txt_file(input_text_path)
|
|
|
17 |
|
18 |
def summarize(self, input_text_path: str, min_length: int):
|
19 |
try:
|
20 |
+
pipe = pipeline("summarization", model="/app/models/bart-large-cnn")
|
21 |
logging.info("Summarization Pipeline Successfully Setup")
|
22 |
|
23 |
text = read_txt_file(input_text_path)
|