import gradio as gr get_completion = pipeline("summarization", model="shleifer/distilbart-cnn-12-6") def summarize(input): output = get_completion(input) return output[0]['summary_text'] iface = gr.Interface(fn=summarize, inputs="input", outputs="text") iface.launch()