sravaniamere commited on
Commit
6b9084f
·
1 Parent(s): b79af36

update ENV_URL to SyncShift space

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -25,7 +25,7 @@ MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
25
  API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY", "dummy")
26
  TASK_NAME = os.getenv("SQL_ENV_TASK", "easy")
27
  BENCHMARK = "sql-correction-env"
28
- ENV_URL = os.getenv("ENV_URL", "http://localhost:7860")
29
  MAX_STEPS = 8
30
  SUCCESS_SCORE_THRESHOLD = 0.5
31
 
@@ -113,7 +113,7 @@ async def run_task(task_name: str) -> None:
113
  log_start(task_name, BENCHMARK, MODEL_NAME)
114
 
115
  try:
116
- reset_resp = await http.post("/reset", json={"difficulty": task_name})
117
  reset_resp.raise_for_status()
118
  obs = reset_resp.json()
119
 
 
25
  API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY", "dummy")
26
  TASK_NAME = os.getenv("SQL_ENV_TASK", "easy")
27
  BENCHMARK = "sql-correction-env"
28
+ ENV_URL = os.getenv("ENV_URL", "https://SyncShift-sql-correction-env.hf.space")
29
  MAX_STEPS = 8
30
  SUCCESS_SCORE_THRESHOLD = 0.5
31
 
 
113
  log_start(task_name, BENCHMARK, MODEL_NAME)
114
 
115
  try:
116
+ reset_resp = await http.post("/reset", json={"task_name": task_name})
117
  reset_resp.raise_for_status()
118
  obs = reset_resp.json()
119