Spaces:
Runtime error
Runtime error
add model and tokenizer args to pipeline
Browse files
app.py
CHANGED
|
@@ -71,7 +71,9 @@ if __name__ == "__main__":
|
|
| 71 |
text="Creating abstractive summary. This might take a few seconds ..."
|
| 72 |
):
|
| 73 |
text_to_summarize = clean_txt
|
| 74 |
-
abs_summarizer = pipeline(
|
|
|
|
|
|
|
| 75 |
if not is_url:
|
| 76 |
# list of chunks
|
| 77 |
text_to_summarize = preprocess_text_for_abstractive_summarization(
|
|
|
|
| 71 |
text="Creating abstractive summary. This might take a few seconds ..."
|
| 72 |
):
|
| 73 |
text_to_summarize = clean_txt
|
| 74 |
+
abs_summarizer = pipeline(
|
| 75 |
+
"summarization", model=abs_model_name, tokenizer=abs_tokenizer_name
|
| 76 |
+
)
|
| 77 |
if not is_url:
|
| 78 |
# list of chunks
|
| 79 |
text_to_summarize = preprocess_text_for_abstractive_summarization(
|