Update app.py

app.py

@@ -1,9 +1,15 @@
 import json, numpy as np, gradio as gr
 from setfit import SetFitModel
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, HfHubHTTPError, EntryNotFoundError
 
 MODEL_ID = "DelaliScratchwerk/text-period-setfit"
-LABELS = json.load(open(hf_hub_download(MODEL_ID, "labels.json")))  # pulls from model repo
+
+# ---- load labels (Hub -> local fallback)
+try:
+    labels_path = hf_hub_download(MODEL_ID, "labels.json")
+    LABELS = json.load(open(labels_path))
+except (HfHubHTTPError, EntryNotFoundError, FileNotFoundError):
+    LABELS = json.load(open("labels.json"))
 
 model = SetFitModel.from_pretrained(MODEL_ID)