Shreeraj Mummidivarapu commited on
Commit
4312101
Β·
unverified Β·
1 Parent(s): 71f11e2

Bhagavan Mera Madad Karo πŸ™

Browse files
Files changed (1) hide show
  1. inference.py +5 -5
inference.py CHANGED
@@ -19,14 +19,14 @@ from openai import OpenAI
19
  # ── Credentials ───────────────────────────────────────────────────────────────
20
  # The hackathon validator INJECTS API_BASE_URL and API_KEY into the environment.
21
  # We MUST use those values directly β€” never override them with HF_TOKEN or defaults.
22
- API_BASE_URL = os.environ.get("API_BASE_URL", "https://router.huggingface.co/v1")
23
- API_KEY = os.environ.get("API_KEY") or os.environ.get("HF_TOKEN")
24
  if not API_KEY:
25
  print("WARNING: API_KEY not set. LLM calls will fail.", file=sys.stderr, flush=True)
26
  API_KEY = "missing"
27
 
28
- MODEL_NAME = os.environ.get("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
29
- ENV_BASE_URL = os.environ.get("ENV_BASE_URL", "https://huggingface.co/spaces/anonymousDevil/cognitive-load-manager")
30
 
31
  print("DEBUG BASE URL:", API_BASE_URL, flush=True)
32
  print("DEBUG MODEL:", MODEL_NAME, flush=True)
@@ -81,7 +81,7 @@ def log_end(success: bool, steps: int, score: float, rewards: List[float]) -> No
81
 
82
  # ── MAIN ───────────────────────────────────────────────────────
83
  def main():
84
- task_id = os.environ.get("CLM_LEVEL", "hard")
85
 
86
  log_start(task=TASK_NAME, env=BENCHMARK, model=MODEL_NAME)
87
 
 
19
  # ── Credentials ───────────────────────────────────────────────────────────────
20
  # The hackathon validator INJECTS API_BASE_URL and API_KEY into the environment.
21
  # We MUST use those values directly β€” never override them with HF_TOKEN or defaults.
22
+ API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
23
+ API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
24
  if not API_KEY:
25
  print("WARNING: API_KEY not set. LLM calls will fail.", file=sys.stderr, flush=True)
26
  API_KEY = "missing"
27
 
28
+ MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
29
+ ENV_BASE_URL = os.getenv("ENV_BASE_URL", "https://huggingface.co/spaces/anonymousDevil/cognitive-load-manager")
30
 
31
  print("DEBUG BASE URL:", API_BASE_URL, flush=True)
32
  print("DEBUG MODEL:", MODEL_NAME, flush=True)
 
81
 
82
  # ── MAIN ───────────────────────────────────────────────────────
83
  def main():
84
+ task_id = os.getenv("CLM_LEVEL", "hard")
85
 
86
  log_start(task=TASK_NAME, env=BENCHMARK, model=MODEL_NAME)
87