text_summarizer / app.py
Oumnia's picture
Create app.py
e6665e0 verified
import torch
import gradio as gr
from transformers import pipeline
text_summary = pipeline("summarization", model= "sshleifer/distilbart-xsum-12-6" )
def summarise_txt(input):
output = text_summary(input)
return output[0]['summary_text']
gr.close_all()
demo = gr.Interface(
fn= summarise_txt,
inputs= [gr.Textbox(label="input a text to summarize in here", lines=8)],
outputs= [gr.Textbox(label= "text after summarization", lines= 4)],
title= "text summarizer project",
description= "the project takes a text as input and a summarized text as an output using hugging face models"
)
demo.launch()