Add sentiment app

app.py

@@ -0,0 +1,17 @@
+import streamlit as st
+from model.inference import predict_sentiment
+
+st.set_page_config(page_title="Sentiment AI", page_icon="💡")
+
+st.title("💡 Sentiment AI Analyzer")
+
+text = st.text_area("Enter your text:")
+
+if st.button("Analyze"):
+    if text.strip() != "":
+        label, confidence = predict_sentiment(text)
+        
+        st.subheader(f"Result: {label}")
+        st.write(f"Confidence: {confidence:.2f}")
+    else:
+        st.warning("Please enter some text")

model/inference.py

@@ -0,0 +1,20 @@
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
+model_name = "Jurk06/bert-uncased-sst2"
+
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+
+labels = ["Negative", "Positive"]
+
+def predict_sentiment(text):
+    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
+    
+    outputs = model(**inputs)
+    probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    
+    pred = torch.argmax(probs).item()
+    confidence = probs[0][pred].item()
+    
+    return labels[pred], confidence

requirements.txt

@@ -1,3 +1,6 @@
 altair
 pandas
+streamlit
+transformers
+torch
 streamlit