| from transformers import pipeline | |
| import gradio as gr | |
| model = pipeline( | |
| "summarization", | |
| model="sshleifer/distilbart-cnn-12-6", | |
| framework="pt" # 👈 force PyTorch | |
| ) | |
| def predict(prompt): | |
| summary = model(prompt)[0]['summary_text'] | |
| return summary | |
| with gr.Blocks() as demo: | |
| textbox = gr.Textbox(placeholder="Enter text block to summarize",lines=4) | |
| gr.Interface(fn=predict,inputs=textbox,outputs='text') | |
| demo.launch() | |