Update app.py

app.py

@@ -1,14 +1,14 @@
 import json, numpy as np, gradio as gr
 from setfit import SetFitModel
-from huggingface_hub import hf_hub_download, HfHubHTTPError, EntryNotFoundError
+from huggingface_hub import hf_hub_download
 
 MODEL_ID = "DelaliScratchwerk/text-period-setfit"
 
-# ---- load labels (Hub -> local fallback)
+# Try to load labels from the model repo; if missing, use local labels.json
 try:
     labels_path = hf_hub_download(MODEL_ID, "labels.json")
     LABELS = json.load(open(labels_path))
-except (HfHubHTTPError, EntryNotFoundError, FileNotFoundError):
+except Exception:
     LABELS = json.load(open("labels.json"))
 
 model = SetFitModel.from_pretrained(MODEL_ID)
@@ -37,5 +37,6 @@ demo = gr.Interface(
     cache_examples=False,
     allow_flagging="never",
 )
+
 if __name__ == "__main__":
     demo.launch()