arpm01 commited on
Commit
83b7249
·
1 Parent(s): 89032c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,9 +4,9 @@ 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="image-classification",
8
  model=model,
9
- tokenizer=tokenizer)
10
 
11
  gr.Interface.from_pipeline(pipe,
12
  title="financial summarization",
 
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,)
10
 
11
  gr.Interface.from_pipeline(pipe,
12
  title="financial summarization",