Update streamlit_app.py

streamlit_app.py

@@ -0,0 +1,41 @@
+import streamlit as st
+from transformers import pipeline
+
+# Load AI model
+classifier = pipeline(
+    "text-classification",
+    model="mrm8488/bert-tiny-finetuned-sms-spam-detection"
+)
+
+# Page config
+st.set_page_config(
+    page_title="AI Cyber Threat Detector",
+    page_icon="🛡️"
+)
+
+st.title("🛡️ AI Cyber Threat Detector")
+
+text = st.text_input(
+    "Enter URL or message"
+)
+
+if st.button("Analyze"):
+
+    if text.strip() == "":
+        st.warning("Please enter text")
+
+    else:
+
+        with st.spinner("Analyzing..."):
+
+            result = classifier(text)[0]
+
+            label = result["label"]
+            confidence = round(result["score"] * 100, 2)
+
+            if label.lower() == "spam":
+                st.error(f"⚠️ Threat Detected: {label}")
+            else:
+                st.success(f"✅ Safe: {label}")
+
+            st.write("Confidence:", confidence, "%")