arpm01 commited on
Commit
0c6a00e
·
1 Parent(s): 2a5d038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -4,6 +4,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
4
  tokenizer = AutoTokenizer.from_pretrained("human-centered-summarization/financial-summarization-pegasus")
5
 
6
  model = AutoModelForSeq2SeqLM.from_pretrained("human-centered-summarization/financial-summarization-pegasus")
 
7
  pipe = pipeline(task="summarization",
8
  model=model,
9
  tokenizer=tokenizer,)
 
4
  tokenizer = AutoTokenizer.from_pretrained("human-centered-summarization/financial-summarization-pegasus")
5
 
6
  model = AutoModelForSeq2SeqLM.from_pretrained("human-centered-summarization/financial-summarization-pegasus")
7
+
8
  pipe = pipeline(task="summarization",
9
  model=model,
10
  tokenizer=tokenizer,)