Update app.py

app.py

@@ -22,11 +22,12 @@ MODEL_CHOICES = [
     "FractalAIResearch/Fathom-R1-14B"  # placeholder — replace with real phantom model
 ]
 
+PIPELINES = {}
 
-PIPELINES = {
-    name: pipeline("zero-shot-classification", model=name)
-    for name in MODEL_CHOICES
-}
+def get_pipeline(name):
+    if name not in PIPELINES:
+        PIPELINES[name] = pipeline("zero-shot-classification", model=name)
+    return PIPELINES[name]
 
 # ————————————————
 # Ensure log files exist
@@ -48,7 +49,7 @@ def run_stage1(question, model_name):
     if not question or not question.strip():
         return {}, gr.update(choices=[]), ""
     start = time.time()
-    clf = PIPELINES[model_name]
+    clf = get_pipeline(model_name)
     out = clf(question, candidate_labels=coarse_labels)
     labels, scores = out["labels"][:3], out["scores"][:3]
     duration = round(time.time() - start, 3)
@@ -71,7 +72,7 @@ def run_stage2(question, model_name, subject):
 
     # 2) Inference (fast, using preloaded pipeline)
     start = time.time()
-    clf = PIPELINES[model_name]
+    clf = get_pipeline(model_name)
     out = clf(question, candidate_labels=fine_labels)
     labels, scores = out["labels"][:3], out["scores"][:3]
     duration = round(time.time() - start, 3)