Update app.py

app.py

@@ -2,7 +2,7 @@ import pickle
 import streamlit as st
 import numpy as np
 
-# --- Load saved model, vectorizer, and binarizer ---
+# Load saved vectorizer, model, and binarizer
 with open("vectorizer (3).pkl", "rb") as f:
     vectorizer = pickle.load(f)
 
@@ -12,37 +12,37 @@ with open("model (6).pkl", "rb") as f:
 with open("binarizer (3).pkl", "rb") as f:
     mlb = pickle.load(f)
 
-# --- Streamlit App UI ---
 st.title("🔖 Stack Overflow Tags Predictor")
 st.markdown("Enter a question title and description. Tags will be predicted automatically based on model confidence.")
 
 title = st.text_input("📌 Enter Question Title")
 description = st.text_area("📝 Enter Question Description", height=150)
 
-# --- Prediction Logic ---
+# 🔧 Adjust this to control how many tags are returned
+threshold = 0.2  # Lower threshold means more tags predicted
+
 def predict_tags_auto(title, description, threshold=0.2):
     input_text = title + " " + description
     input_vector = vectorizer.transform([input_text])
 
-    # Get class probabilities from model
-    probas_list = model.predict_proba(input_vector)  # List of arrays
+    # Get probabilities for each tag
+    probas = model.predict_proba(input_vector)
 
-    # Convert list of (1, n_classes) to shape (1, n_classes)
-    proba_matrix = np.hstack([p[:, 1] if p.shape[1] == 2 else p[:, 0] for p in probas_list])
+    # Combine probabilities from each classifier
+    probas_array = np.array([class_proba[0] for class_proba in probas])  # shape: (n_classes,)
 
     # Apply threshold
-    predicted_binary = (proba_matrix >= threshold).astype(int).reshape(1, -1)
+    predicted_binary = (probas_array >= threshold).astype(int).reshape(1, -1)
 
-    # Decode binary predictions to tag labels
+    # Convert binary to tags
     tags = mlb.inverse_transform(predicted_binary)
-    return tags[0] if tags and len(tags[0]) > 0 else []
+    return tags[0] if tags else []
 
-# --- Predict Button ---
 if st.button("Predict Tags"):
     if not title.strip() or not description.strip():
         st.warning("⚠️ Please enter both title and description.")
     else:
-        tags = predict_tags_auto(title, description, threshold=0.4)
+        tags = predict_tags_auto(title, description, threshold)
         if tags:
             st.success("✅ Predicted Tags: " + ", ".join(tags))
         else: