Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import PegasusForConditionalGeneration | |
from transformers import PegasusTokenizer | |
from transformers import pipeline | |
model_name = "google/pegasus-xsum" | |
pegasus_tokenizer = PegasusTokenizer.from_pretrained(model_name) | |
def summarize(input_text): | |
nwords=len(input_text.split(" ")) | |
# Define summarization pipeline | |
summarizer = pipeline("summarization", model=model_name, tokenizer=pegasus_tokenizer,min_length=int(nwords/10)+10, max_length=int(nwords/5+10), framework="pt") | |
summary=summarizer(input_text)[0]['summary_text'] | |
return(summary) | |
gr.Interface(fn=summarize,inputs=gr.inputs.Textbox(placeholder="Paste the text to be summarized here..."),outputs="textbox").launch(); |