Spaces:
Runtime error
Runtime error
import gradio as gr | |
from setfit import AbsaModel | |
# Load the ABSA model | |
model = AbsaModel.from_pretrained( | |
"pahri/setfit-indo-restomix-aspect", | |
"pahri/setfit-indo-restomix-polarity", | |
spacy_model="id_core_news_trf", | |
) | |
def analyze_text(text): | |
""" | |
Analyzes the input text using the ABSA model and returns aspects and sentiment. | |
Args: | |
text: The text to be analyzed. | |
Returns: | |
A formatted string containing aspects and sentiment. | |
""" | |
pred = model.predict(text) | |
return pred | |
title = "Analisa Review Restoran Anda" | |
examples = [["Penyajiannya super aesthetic & instagramable.. Tempatnya juga keren, bikin betah & pengen foto-foto terus. Pelayanan ramah, parkir lumayan memadai. Recommended"]] | |
description = "Masukkan review restoran Anda untuk menganalisa aspek dan sentimen." | |
interface = gr.Interface( | |
fn=analyze_text, | |
inputs=gr.Textbox(lines=2, placeholder="Masukkan review restoran Anda di sini..."), | |
outputs=gr.Text(), | |
description=description, | |
title=title, | |
examples=examples | |
) | |
interface.launch() | |