from transformers import pipeline
import gradio as gr
import html

summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")

def translate(text):
    # HTMLエンティティを一括でデコード
    text = html.unescape(text)
    # 長文入力でクラッシュしないように max_length を設定
    result = summarizer(text, min_length=80, max_length=256, truncation=True)
    return result[0]["summary_text"]

iface = gr.Interface(
    fn=translate,
    input=gr.inputs.Textbox(lines=10,placeholder="Enter your text to summarize..."),
    outputs="text"
)
iface.launch()