Update app.py

app.py

@@ -3,6 +3,7 @@ import pickle
 import re
 import numpy as np
 
+
 # Streamlit page configuration
 st.set_page_config(page_title="Stack Overflow Tags Predictor", layout="centered")
 
@@ -43,8 +44,14 @@ if st.button("🔍 Predict Tags"):
         input_text = clean_text(title + " " + body)
         X_input = vectorizer.transform([input_text])
 
-        # Predict using model
-        y_pred = model.predict(X_input)
+        try:
+            # Use predict_proba and apply fixed threshold
+            y_prob = model.predict_proba(X_input)
+            y_pred = (y_prob >= 0.3).astype(int)  # Default threshold used
+        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]:
@@ -52,3 +59,4 @@ if st.button("🔍 Predict Tags"):
             st.write(", ".join(predicted_tags[0]))
         else:
             st.info("🤔 No tags predicted. Try refining your question.")
+