from transformers import pipeline
import gradio as gr

pipe = pipeline("text-generation", model="Sigurdur/icebreaker")

demo = gr.Interface.from_pipeline(pipe)
demo.launch()