Spaces:
Runtime error
Runtime error
File size: 559 Bytes
6df46b0 01f683c 6df46b0 a7486bd 6df46b0 01f683c 6df46b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
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()
|