Pathberries/Bart-Large-CNN-API
0
1import gradio as gr2from transformers import pipeline3import torch4 5torch.set_default_dtype(torch.float32)6 7summarizer = pipeline("summarization", model="facebook/bart-large-cnn")8print("Model Loaded")9 10def summarize(text):11 return (summarizer(text, max_length=300, min_length=200, do_sample=False, num_workers=4))12 13iface = gr.Interface(fn=summarize, inputs="text", outputs="text")14iface.launch(debug=True)15 