File size: 257 Bytes
082a05e
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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()