Shreeraj Mummidivarapu commited on
Commit
0d60edf
·
unverified ·
1 Parent(s): 1432328

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -17,9 +17,9 @@ from openai import OpenAI
17
  # We try HF_TOKEN first (same order as the accepted reference repo).
18
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
19
  MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
20
- HF_TOKEN = os.getenv("HF_TOKEN") # auto-injected by HF Space
21
- API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY") # try HF_TOKEN first
22
- ENV_BASE_URL = os.getenv("ENV_BASE_URL", "http://localhost:7860")
23
 
24
  TASK_NAME = "schedule-optimization"
25
  BENCHMARK = "cognitive-load-manager"
 
17
  # We try HF_TOKEN first (same order as the accepted reference repo).
18
  API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
19
  MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
20
+ HF_TOKEN = os.getenv("HF_TOKEN")
21
+ API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
22
+ ENV_BASE_URL = os.getenv("ENV_BASE_URL", "https://anonymousdevil-cognitive-load-manager.hf.space")
23
 
24
  TASK_NAME = "schedule-optimization"
25
  BENCHMARK = "cognitive-load-manager"