backend/sarcasm.py

import streamlit as st
from .sa import predictor


def write():
    st.markdown(
        """
        # Arabic Sarcasm Detection

        This is a simple sarcasm detection app that uses the [MARBERT](https://huggingface.co/UBC-NLP/MARBERT) model trained on [ArSarcasm](https://github.com/iabufarha/ArSarcasm)
        """
    )

    input_text = st.text_input(
        "Enter your text here:",
    )
    if st.button("Predict"):
        with st.spinner("Predicting..."):
            prediction, scores = predictor.get_preds_from_sarcasm([input_text])
            st.write(f"Result: {prediction[0]}")
            st.write(f"Score: {scores[0]}")