NazmusAshrafi's picture
Added reqs
ebb742b
raw
history blame
540 Bytes
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")