Milestone3begin

app.py

@@ -2,8 +2,17 @@ import streamlit as st
 import tensorflow as tf
 from transformers import pipeline
 from textblob import TextBlob
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+import torch.nn.functional as F
 
-classifier = pipeline(task="sentiment-analysis")
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+classifier = pipeline(task="sentiment-analysis", model=model, tokenizer=tokenizer)
+# classifier = pipeline(task="sentiment-analysis")
 
 textIn = st.text_input("Input Text Here:", "I really like the color of your car!")