Gladiator commited on
Commit
c099517
1 Parent(s): 121b578

add model and tokenizer args to pipeline

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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("summarization")
 
 
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(