import streamlit as st
from transformers import pipeline
st.title(" :smile: IMDB MOVIES SENTIMENT ANALYSIS")

user_input = st.text_input("Enter your text:")

classifier=pipeline("text-classification",model="flexnotop/imdb-v1",device=0)
if st.button("Classify"):
    if user_input:
        result=classifier(user_input)
        if result[0]['label']=="LABEL_1":
            st.write("Positive Rating")
        else:
            st.write("Negative Rating")
    else:
        st.write("Please enter some text")