Update app.py

app.py

@@ -1,16 +1,15 @@
 import streamlit as st
 import pickle
-import numpy as np
 
-# ⚙️ Set Streamlit Page Config (this MUST be first)
-st.set_page_config(page_title="Multi-Tag Stack Overflow Predictor", page_icon="🧠")
+# ✅ Set Streamlit page config (must be first command)
+st.set_page_config(page_title="🔖 Stack Overflow Multi-Tag Predictor", page_icon="🧠")
 
-# 🧪 Load Model Files
+# 🔃 Load model files
 try:
     with open("vectorizer.pkl", "rb") as f:
         vectorizer = pickle.load(f)
 
-    with open("model (2).pkl", "rb") as f:
+    with open("model.pkl", "rb") as f:
         model = pickle.load(f)
 
     with open("binarizer.pkl", "rb") as f:
@@ -20,40 +19,33 @@ except Exception as e:
     st.error(f"❌ Error loading model files: {str(e)}")
     st.stop()
 
-# 🧠 Prediction Function (Multi-tag with Threshold)
-def predict_tags(title, description, threshold=0.3):
+# 🧠 Prediction function
+def predict_tags(title, description):
     try:
         if not title.strip() or not description.strip():
             return "⚠️ Please enter both title and description."
 
         input_text = title + " " + description
         input_vector = vectorizer.transform([input_text])
+        prediction = model.predict(input_vector)
 
-        # Predict probabilities
-        probas = model.predict_proba(input_vector)
-
-        # Apply threshold to get binary predictions
-        multi_pred = (probas >= threshold).astype(int)
-
-        # Convert binary predictions to tags
-        predicted_tags = mlb.inverse_transform(multi_pred)
+        predicted_tags = mlb.inverse_transform(prediction)
 
         if predicted_tags and predicted_tags[0]:
             return "✅ **Predicted Tags:** " + ", ".join(predicted_tags[0])
         else:
-            return "ℹ️ No tags predicted. Try lowering the threshold or refining your input."
+            return "ℹ️ No tags predicted. Try refining your question."
 
     except Exception as e:
         return f"❌ Error during prediction: {str(e)}"
 
 # 🚀 Streamlit UI
-st.title("🔖 Stack Overflow Multi-Tag Predictor")
-st.markdown("Enter a question title and description to predict relevant tags using ML.")
+st.title("💬 Stack Overflow Multi-Tag Predictor")
+st.markdown("Enter a question title and description to get multiple relevant tags.")
 
-title = st.text_input("📌 Question Title")
-description = st.text_area("📝 Question Description", height=150)
-threshold = st.slider("🎯 Prediction Threshold", 0.1, 0.9, 0.3, step=0.05)
+title = st.text_input("📌 Enter Question Title")
+description = st.text_area("📝 Enter Question Description", height=150)
 
-if st.button("Predict Tags"):
-    result = predict_tags(title, description, threshold)
+if st.button("🎯 Predict Tags"):
+    result = predict_tags(title, description)
     st.markdown(result)