Spaces:
Sleeping
Sleeping
Commit ·
6b9084f
1
Parent(s): b79af36
update ENV_URL to SyncShift space
Browse files- 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
|
| 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={"
|
| 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 |
|