ayoubkirouane's picture
Update app.py
d6bb89e verified
raw
history blame contribute delete
568 Bytes
from transformers import pipeline
import gradio as gr
pipe = pipeline("summarization", model="ayoubkirouane/T5-4-Summarization")
def summarization(text):
return pipe(text)[0]["summary_text"]
# Create a Gradio interface
iface = gr.Interface(
fn=summarization,
inputs=gr.Textbox(label="Input Text"),
outputs=gr.Textbox(label="Generated Summary"),
allow_flagging=False,
title="T5-4-Summarization",
description="This app generates a summary of the input text using T5 fine-tuned model.",
)
# Launch the Gradio app
iface.launch(debug=True)