File size: 1,108 Bytes
7a31ba6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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"