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