import gradio as gr
from transformers import pipeline

classifier = pipeline(task='text-classification', model='cwchang/my-awesome-tx-v2', device=-1)

def classify(text):
    return classifier(text)[0]["label"]

demo = gr.Interface(
    fn=classify,
    inputs=gr.Textbox(placeholder="Please enter the text..."),
    outputs="label",
    examples=[
        ["What's the weather like today?"],
        ["Set an alarm for 7 AM tomorrow"],
        ["Call Mom"],
        ["Send a text to Alex saying, 'I'll be there in 15 minutes'"],
        ["Play some relaxing music"],
        ["Remind me to buy milk when I'm at the grocery store"],
        ["How do I get to the nearest coffee shop?"],
        ["What's the latest news?"],
        ["Translate 'thank you' into Spanish"],
        ["Add a meeting to my calendar for next Monday at 3 PM"]]
    )

demo.launch()