surajit2839 commited on
Commit
908cd92
·
verified ·
1 Parent(s): bfa7a07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +42 -0
app.py CHANGED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # --- PAGE SETUP ---
5
+ st.set_page_config(page_title="Hugging Face Sentiment AI", page_icon="🤗")
6
+
7
+ # --- MODEL LOADING (Cached) ---
8
+ @st.cache_resource
9
+ def load_sentiment_model():
10
+ # This uses the default 'distilbert-base-uncased-finetuned-sst-2-english'
11
+ return pipeline("sentiment-analysis")
12
+
13
+ classifier = load_sentiment_model()
14
+
15
+ # --- UI ELEMENTS ---
16
+ st.title("🤗 AI Sentiment Analyzer")
17
+ st.write("This app uses a Hugging Face Transformer model to detect sentiment.")
18
+
19
+ user_input = st.text_area("Enter text to analyze:", placeholder="I am so excited to build this app!")
20
+
21
+ if st.button("Analyze Sentiment"):
22
+ if user_input.strip():
23
+ with st.spinner("Analyzing..."):
24
+ # Run prediction
25
+ results = classifier(user_input)
26
+
27
+ # Extract data
28
+ label = results[0]['label']
29
+ score = results[0]['score']
30
+
31
+ # --- DISPLAY RESULTS ---
32
+ st.divider()
33
+ if label == "POSITIVE":
34
+ st.success(f"### {label} 😊")
35
+ else:
36
+ st.error(f"### {label} 😡")
37
+
38
+ st.metric(label="Confidence Score", value=f"{score:.2%}")
39
+ else:
40
+ st.warning("Please enter some text first!")
41
+
42
+ st.caption("Running on Hugging Face Spaces with Streamlit")