aaditya's picture
Update src/envs.py
9862d77 verified
raw
history blame
918 Bytes
import os
from huggingface_hub import HfApi
H4_TOKEN = os.environ.get("HF_SECRET", None)
REPO_ID = "openlifescienceai/open_medical_llm_leaderboard"
QUEUE_REPO = "openlifescienceai/test_requests"
RESULTS_REPO = "openlifescienceai/test_results"
# have not created these repos yet
PRIVATE_QUEUE_REPO = "openlifescienceai/test_private-requests"
PRIVATE_RESULTS_REPO = "openlifescienceai/test_private-results"
IS_PUBLIC = bool(os.environ.get("IS_PUBLIC", True))
CACHE_PATH = os.getenv("HF_HOME", ".")
print(f"CACHE_PATH = {CACHE_PATH}")
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
EVAL_REQUESTS_PATH_PRIVATE = "eval-queue-private"
EVAL_RESULTS_PATH_PRIVATE = "eval-results-private"
# Rate limit variables
RATE_LIMIT_PERIOD = 7
RATE_LIMIT_QUOTA = 5
HAS_HIGHER_RATE_LIMIT = ["TheBloke"]
API = HfApi(token=H4_TOKEN)
# API = HfApi()