app.py

from transformers import pipeline
import gradio as gr

model = pipeline("text-generation", model="ButteredToast/ToastBot")

def generate_text(input_text):
    return model(input_text)[0]["generated_text"]

demo = gr.Interface(fn=generate_text, inputs="text", outputs="text")
demo.launch()