json file handling

qwen_classifier/evaluate.py

@@ -10,14 +10,20 @@ import torch
 from datasets import Dataset
 from torch.utils.data import DataLoader
 import requests
+from pathlib import Path
 
 from .config import TAG_NAMES, DEVICE, SPACE_URL, EVAL_LIMIT
 from .globals import global_model, global_tokenizer
 
+
 def _load_data(test_data_path):
+    test_data_path = Path(__file__).parent / test_data_path
     # Check file existence
     if not os.path.exists(test_data_path):
-        raise FileNotFoundError(f"Zip file not found at: {test_data_path}")
+        raise FileNotFoundError(
+            f"ZIP file not found at {test_data_path}. "
+            f"Current directory: {os.listdir(Path(__file__).parent)}"
+        )
     
     if not zipfile.is_zipfile(test_data_path):
         raise zipfile.BadZipFile(f"File is not a valid zip archive: {test_data_path}")

setup.py

@@ -4,6 +4,10 @@ setup(
     name="qwen_classifier",
     version="0.1",
     packages=find_packages(),
+    package_data={
+        'qwen_classifier': ['test_jsons.zip'],
+    },
+    include_package_data=True,
     install_requires=[
         'torch',
         'transformers',