import gradio as gr
from setfit import SetFitModel
model = SetFitModel.from_pretrained("johnpaulbin/toki-pona-classifier-v2")


def greet(name):
    return model.predict([name])[0]

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