File size: 588 Bytes
c2cbc53
5235ed9
8142bba
89032c4
8142bba
89032c4
86b591d
fc7c04c
83b7249
8142bba
a425649
86b591d
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="text2text-generation", 
                model=model,
                tokenizer=tokenizer,)

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