Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| def main(article, min_len, max_len): | |
| summarizer = pipeline("summarization", model="facebook/bart-large-cnn") | |
| return summarizer(article, max_length=int(max_len), min_length=int(min_len), do_sample=False)[0]["summary_text"] | |
| iface = gr.Interface(fn=main, | |
| inputs=[gr.inputs.Textbox(lines=5, label="Article text"), | |
| "number", | |
| "number" | |
| ], | |
| outputs="text") | |
| iface.launch() | |