Update app.py

app.py

@@ -1,10 +1,38 @@
+# import streamlit as st
+# from transformers import pipeline
+
+# sentiment_analysis = pipeline("sentiment-analysis")
+
+# text = st.text_input("Enter some text")
+
+# if text:
+#     result = sentiment_analysis(text)
+#     st.json(result)
+
+
+
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
+
+# Load sentiment analysis model from Hugging Face
+model_name = "distilbert-base-uncased-finetuned-sst-2-english"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
-sentiment_analysis = pipeline("sentiment-analysis")
+# Streamlit UI
+st.title("Sentiment Analysis App")
 
-text = st.text_input("Enter some text")
+# User input
+user_input = st.text_input("Enter a sentence:")
 
-if text:
-    result = sentiment_analysis(text)
-    st.json(result)
+if user_input:
+    # Perform sentiment analysis
+    results = sentiment_analyzer(user_input)
+    
+    # Display sentiment and confidence
+    sentiment = results[0]['label']
+    confidence = results[0]['score']
+    
+    st.write(f"Sentiment: {sentiment}")
+    st.write(f"Confidence: {confidence:.2f}")