Spaces:
Sleeping
Sleeping
File size: 337 Bytes
c7a74ff 99aedab c7a74ff 36999e5 99aedab c7a74ff |
1 2 3 4 5 6 7 8 9 10 11 |
import gradio as gr
from setfit import SetFitModel
def greet(text):
model = SetFitModel.from_pretrained("skylord/setfit-bge-small-v1.5-sst2-8-shot-talk2loop") # Load from the Hugging Face Hub
preds = model.predict([text])
return "Tag: " + preds
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch() |