rickxzo's picture
Update app.py
44df7ce verified
raw
history blame contribute delete
554 Bytes
import streamlit as st
from transformers import pipeline
# Load the custom model and tokenizer
pipe = pipeline("sentiment-analysis", model="rickxzo/review-sentiment-analysis", tokenizer="rickxzo/review-sentiment-analysis")
# Streamlit UI
st.title("Sentiment Analysis with Transformer Model")
text = st.text_area("Enter some text:")
# Perform sentiment analysis when text is provided
if text:
out = pipe(text)
st.json(out)
st.write("### Label Definitions:")
st.write("- **LABEL_2** : Positive")
st.write("- **LABEL_1** : Negative")