Update app.py

app.py

@@ -1,10 +1,9 @@
 import json, numpy as np, gradio as gr
 from setfit import SetFitModel
+from huggingface_hub import hf_hub_download
 
 MODEL_ID = "DelaliScratchwerk/text-period-setfit"
-
-with open("labels.json") as f:
-    LABELS = json.load(f)
+LABELS = json.load(open(hf_hub_download(MODEL_ID, "labels.json")))  # pulls from model repo
 
 model = SetFitModel.from_pretrained(MODEL_ID)
 
@@ -13,8 +12,7 @@ def predict(txt: str):
         return "—", {"error": "Please paste some text."}
     probs = np.asarray(model.predict_proba([txt])[0], dtype=float).ravel()
     if probs.size != len(LABELS):
-        # hard stop so you notice a mismatch
-        return "—", {"error": f'label mismatch: model has {probs.size} classes, labels.json has {len(LABELS)}'}
+        return "—", {"error": f"label mismatch: model has {probs.size}, labels.json has {len(LABELS)}"}
     order = np.argsort(probs)[::-1]
     top = LABELS[int(order[0])]
     return top, {LABELS[int(i)]: float(probs[int(i)]) for i in order}