paza-bench / src /envs.py
muchai-mercy's picture
update pazabench space
53a73e0
import os
from dotenv import load_dotenv
from huggingface_hub import HfApi
load_dotenv()
# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("HF_TOKEN")
OWNER = os.environ.get("OWNER")
# ----------------------------------
REPO_ID = os.environ.get("REPO_ID")
QUEUE_REPO = os.environ.get("SUBMISSIONS_REPO")
RESULTS_REPO = os.environ.get("RESULTS_REPO")
CACHE_PATH=os.getenv("HF_HOME", ".")
# Local caches
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
EVAL_REQUESTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-queue-bk")
EVAL_RESULTS_PATH_BACKEND = os.path.join(CACHE_PATH, "eval-results-bk")
# Only create API if token is available
API = HfApi(token=TOKEN) if TOKEN else None
# Flag to enable/disable HuggingFace Hub features
HF_ENABLED = TOKEN is not None and os.environ.get("DISABLE_HF_HUB", "").lower() != "true"
# Telemetry: Set TELEMETRY_ENABLED=true to enable Azure Application Insights
TELEMETRY_ENABLED = os.environ.get("TELEMETRY_ENABLED", "false").lower() == "true"