File size: 585 Bytes
c2cbc53
89032c4
8142bba
89032c4
8142bba
89032c4
83b7249
fc7c04c
83b7249
8142bba
a425649
89032c4
8142bba
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import gradio as gr
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline

tokenizer = AutoTokenizer.from_pretrained("human-centered-summarization/financial-summarization-pegasus")

model = AutoModelForSeq2SeqLM.from_pretrained("human-centered-summarization/financial-summarization-pegasus")
pipe = pipeline(task="summarization", 
                model=model,
                tokenizer=tokenizer,)

gr.Interface.from_pipeline(pipe, 
                    title="financial summarization",
                    description="Add description",
                    ).launch()