pahri's picture
Update app.py
dfffd2c verified
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()