Update main.py

main.py

@@ -16,6 +16,7 @@ import random
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from collections import defaultdict
 from pathlib import Path
+from huggingface_hub import hf_hub_download
 
 # Global model and tokenizer
 model = None
@@ -204,7 +205,14 @@ def load_dataset():
         
 
     
-    dataset_path = Path("xlam_function_calling_60k_processed_with_ground_truth.json")
+    DATASET_FILE = "xlam_function_calling_60k_processed_with_ground_truth.json"
+    dataset_path = Path(
+        hf_hub_download(
+            repo_type="space",
+            repo_id="daoqm123/llm-error-classifier-api",
+            filename=DATASET_FILE,
+        )
+    )
     
     if not dataset_path:
         print("Warning: Dataset file not found. Using hardcoded examples.")