Added the streamlit app

README.md

@@ -1,3 +1,14 @@
+---
+title: Sentiment Analysis App
+emoji: 💻
+colorFrom: gray
+colorTo: red
+sdk: streamlit
+sdk_version: 1.17.0
+app_file: app.py
+pinned: false
+---
+
 # cs-gy-6613-project
 
 ## Installation steps

app.py

@@ -0,0 +1,17 @@
+# Import statements
+import streamlit as st
+from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
+# Get Input
+models = {"Distilbert": "distilbert-base-uncased-finetuned-sst-2-english", "roBERTa" : "cardiffnlp/twitter-roberta-base-sentiment", "Bert" : "finiteautomata/bertweet-base-sentiment-analysis"}
+text = st.text_area('Enter some text!')
+model_name = st.selectbox("What model do you wanna use?",  ("Distilbert", "roBERTa", "Bert"))
+
+# Default Model - DistilBert      
+if text and model_name:
+  model = AutoModelForSequenceClassification.from_pretrained(models[model_name])
+  tokenizer = AutoTokenizer.from_pretrained(models[model_name])
+  classifier = pipeline('sentiment-analysis', model = model, tokenizer = tokenizer)
+  out = classifier(text)
+  st.json(out)

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+torch
+transformers