Shreeraj Mummidivarapu commited on
Bhagavan Mera Madad Karo π
Browse files- 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.
|
| 23 |
-
API_KEY = os.
|
| 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.
|
| 29 |
-
ENV_BASE_URL = os.
|
| 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.
|
| 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 |
|