Added an option to override default tags

app.py

@@ -1,27 +1,41 @@
+# app.py
+import json
 import gradio as gr
 from transformers import pipeline
 
-# 1) Initialize zero-shot classifier
+# 1) Load base labels from JSON
+with open("labels.json", "r") as f:
+    base_labels = json.load(f)
+
+# 2) Prepare default textbox value
+default_label_str = ", ".join(base_labels)
+
+# 3) Initialize zero-shot classifier
 classifier = pipeline(
     task="zero-shot-classification",
     model="facebook/bart-large-mnli"
 )
 
-# 2) Define candidate labels
-LABELS = ["linear algebra", "calculus", "probability", "geometry"]
-
-# 3) Gradio interface function
-def tag_question(question):
-    result = classifier(question, candidate_labels=LABELS)
-    return {lbl: round(score, 3) for lbl, score in zip(result["labels"], result["scores"])}
+# 4) Interface function that merges runtime labels
+def tag_question(question: str, labels_str: str):
+    # Split & clean the user-supplied string
+    labels = [lbl.strip() for lbl in labels_str.split(",") if lbl.strip()]
+    # Zero-shot classify
+    out = classifier(question, candidate_labels=labels)
+    # Return top-3 labels with scores
+    return {lbl: round(score,3) for lbl, score in zip(out["labels"], out["scores"])}
 
-# 4) Build UI
+# 5) Build the Gradio UI
 iface = gr.Interface(
     fn=tag_question,
-    inputs=gr.Textbox(lines=3, placeholder="Enter your MCQ here..."),
+    inputs=[
+        gr.Textbox(lines=3, label="Question"),
+        gr.Textbox(lines=2, label="Candidate Labels (comma-separated)",
+                   value=default_label_str)
+    ],
     outputs=gr.Label(num_top_classes=3),
-    title="Zero-Shot Question Tagger",
-    description="Classify questions into math topics without any training data."
+    title="Hybrid Zero-Shot Question Tagger",
+    description="Loaded labels from `labels.json`, editable at runtime."
 )
 
 if __name__ == "__main__":