hamzabouajila's picture
feat: enhance evaluation queue reliability and add stale job recovery
2f1e30c
raw
history blame contribute delete
912 Bytes
import os
from huggingface_hub import HfApi
# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("HF_TOKEN") # A read/write token for your org
OWNER = "hamzabouajila" # Change to your org - don't forget to create a results and request dataset, with the correct format!
# ----------------------------------
REPO_ID = f"{OWNER}/TunisianLLMLeaderBoard"
QUEUE_REPO = f"{OWNER}/requests"
RESULTS_REPO = f"{OWNER}/results"
# If you setup a cache later, just change HF_HOME
# Local caches
EVAL_REQUESTS_PATH = "./eval-queue"
EVAL_RESULTS_PATH = "./eval-results"
EVAL_REQUESTS_PATH_BACKEND = "./eval-queue-bk"
EVAL_RESULTS_PATH_BACKEND = "./eval-results-bk"
# Create directories if they don't exist
for path in [EVAL_REQUESTS_PATH, EVAL_RESULTS_PATH, EVAL_REQUESTS_PATH_BACKEND, EVAL_RESULTS_PATH_BACKEND]:
os.makedirs(path, exist_ok=True)
API = HfApi(token=TOKEN)