Create app.py

app.py

@@ -0,0 +1,46 @@
+import gradio as gr
+import pickle
+
+# Custom tokenizer (same one used during training)
+def custom_tokenizer(text):
+    return text.lower().split()
+
+# Load artifacts
+with open("model.pkl", "rb") as f:
+    model = pickle.load(f)
+
+with open("tfidf_vectorizer.pkl", "rb") as f:
+    vectorizer = pickle.load(f)
+
+with open("mlb.pkl", "rb") as f:
+    mlb = pickle.load(f)
+
+# Prediction function
+def predict_tags(title, description):
+    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)
+    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 refining your question."
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=predict_tags,
+    inputs=[
+        gr.Textbox(label="Enter Question Title"),
+        gr.Textbox(label="Enter Question Description", lines=4)
+    ],
+    outputs="text",
+    title="🔖 Stack Overflow Tags Predictor",
+    description="Enter a question's title and description to get tag suggestions.",
+    theme="default"
+)
+
+iface.launch()