data1 / app.py
AliUsama98's picture
Update app.py
84213e0
raw
history blame contribute delete
765 Bytes
import gradio as gr
pip install transformers
from transformers import pipeline
# Load the summarization pipeline
summarizer = pipeline("summarization", model="t5-small", tokenizer="t5-small")
# Function for text summarization
def summarize_text(text):
if text:
summary = summarizer(text, max_length=150, min_length=30, do_sample=False)
return summary[0]['summary_text']
return "Please enter text to summarize."
# Create a Gradio interface
iface = gr.Interface(
fn=summarize_text,
inputs=gr.inputs.Textbox(lines=10, label="Enter text to summarize"),
outputs=gr.outputs.Textbox(label="Summary"),
title="Text Summarization",
description="Use the model to summarize text."
)
# Launch the Gradio interface
iface.launch()