Spaces:
Sleeping
Sleeping
Commit ·
13f248e
1
Parent(s): be7b1e3
fixed inference
Browse files- inference.py +3 -0
inference.py
CHANGED
|
@@ -26,6 +26,9 @@ if HF_TOKEN is None:
|
|
| 26 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 27 |
|
| 28 |
API_KEY = HF_TOKEN
|
|
|
|
|
|
|
|
|
|
| 29 |
# ─────────────────────────────────────────────
|
| 30 |
|
| 31 |
def log_start(task: str, env: str, model: str) -> None:
|
|
|
|
| 26 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 27 |
|
| 28 |
API_KEY = HF_TOKEN
|
| 29 |
+
BENCHMARK = "sql-query-debugger"
|
| 30 |
+
MAX_STEPS = 10
|
| 31 |
+
SUCCESS_SCORE_THRESHOLD = 0.5
|
| 32 |
# ─────────────────────────────────────────────
|
| 33 |
|
| 34 |
def log_start(task: str, env: str, model: str) -> None:
|