Update app.py

app.py

@@ -3,7 +3,6 @@ import pickle
 import re
 import numpy as np
 
-
 # Streamlit page configuration
 st.set_page_config(page_title="Stack Overflow Tags Predictor", layout="centered")
 
@@ -36,9 +35,6 @@ st.markdown("Enter a question's *title* and *description*, and this app will sug
 title = st.text_input("📝 Question Title")
 body = st.text_area("📄 Question Description", height=200)
 
-# Optional: Add a threshold slider
-#threshold = st.slider("🔧 Tag Confidence Threshold", min_value=0.1, max_value=0.9, value=0.3, step=0.05)
-
 # Prediction Button
 if st.button("🔍 Predict Tags"):
     if not title.strip() or not body.strip():
@@ -48,18 +44,17 @@ if st.button("🔍 Predict Tags"):
         X_input = vectorizer.transform([input_text])
 
         try:
-            # Use predict_proba and apply threshold
-            y_prob = model.predict_proba(X_input)
-            y_pred = (y_prob >= threshold).astype(int)
-        except AttributeError:
-            st.warning("⚠ Model does not support `predict_proba`. Using default `predict` method.")
             y_pred = model.predict(X_input)
-
-        predicted_tags = mlb.inverse_transform(y_pred)
-
-        if predicted_tags and predicted_tags[0]:
-            st.success("✅ Predicted Tags:")
-            st.write(", ".join(predicted_tags[0]))
-        else:
-            st.info("🤔 No tags predicted. Try refining your question or lowering the threshold.")
+        except Exception as e:
+            st.error(f"❌ Prediction failed: {e}")
+            y_pred = None
+
+        if y_pred is not None:
+            predicted_tags = mlb.inverse_transform(y_pred)
+
+            if predicted_tags and predicted_tags[0]:
+                st.success("✅ Predicted Tags:")
+                st.write(", ".join(predicted_tags[0]))
+            else:
+                st.info("🤔 No tags predicted. Try refining your question.")