app.py

import streamlit as st
from transformers import pipeline
import utils

st.title("Indo LEGO-ABSA")
example_text = "EXAMPLE: Pizza nya enak sekali"
text = st.text_area("Enter your text here", "", placeholder=example_text)
aspect = st.checkbox("Aspect", value=True)
opinion = st.checkbox("Opinion")
sentiment = st.checkbox("Sentiment")
pipe = pipeline(model="rdyzakya/IndoLEGO-ABSA")
pipe.model.config.max_length = 128
prompt = ""
se_order = ""

if text:
    if aspect:
        se_order += 'a'
    if opinion:
        se_order += 'o'
    if sentiment:
        se_order += 's'
    se_order = set(se_order)
    if not se_order.issubset("acos"):
        st.json([])
    else:
        se_order = sorted(list(se_order))
        se_order = ''.join(se_order)
        inputs = utils.add_prompt(text, se_order)
        out = pipe(inputs)
        result = utils.catch_answer(out[0]["generated_text"], se_order)
        st.json(result)