| import gradio as gr | |
| from setfit import AbsaModel | |
| model = AbsaModel.from_pretrained( | |
| "tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-aspect", | |
| "tomaarsen/setfit-absa-paraphrase-mpnet-base-v2-restaurants-polarity" | |
| ) | |
| def predict(text): | |
| return model.predict(text) | |
| demo = gr.Interface(fn=predict, inputs="text", outputs="text") | |
| demo.launch() |