import streamlit as st
from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer

tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")

pulled_model = AutoModelForSequenceClassification.from_pretrained(
    "NazmusAshrafi/stock_twitter_sentiment_Bert")

classifier = pipeline("sentiment-analysis",
                      model=pulled_model, tokenizer=tokenizer)


print(classifier("AAP Up Theory = Will go up in future, buy now"))

st.title("Topic Modeling and Sentiment Analysis of Stock Tweets")