Upload folder using huggingface_hub

inference.py

@@ -29,7 +29,7 @@ from openai import OpenAI
 
 # ── Configuration ──────────────────────────────────────────────────────────
 API_BASE_URL   = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
-MODEL_NAME     = os.getenv("MODEL_NAME",   "Qwen/Qwen2.5-72B-Instruct")
+MODEL_NAME     = os.getenv("MODEL_NAME",   "Qwen/Qwen2.5-7B-Instruct")
 API_KEY        = os.getenv("HF_TOKEN") or os.getenv("OPENAI_API_KEY", "")
 IMAGE_NAME     = os.getenv("LOCAL_IMAGE_NAME", "nl2sql-bench:latest")
 SPACE_URL      = os.getenv("SPACE_URL", "http://localhost:8000")
@@ -171,7 +171,7 @@ async def run_task(client: OpenAI, env, task_name: str) -> dict:
         # OpenEnv reset() may not accept task args via HTTP; we rely on
         # NL2SQL_DEFAULT_TASK env-var being set before calling, OR we
         # pass it as a reset parameter if the server supports it.
-        result = await env.reset()
+        result = await env.reset(options={"task": task_name}) # changed
         obs = result.observation
 
         for step in range(1, MAX_STEPS + 1):
@@ -191,7 +191,7 @@ async def run_task(client: OpenAI, env, task_name: str) -> dict:
             sql = call_llm(client, user_prompt)
 
             from client import NL2SQLAction  # local to avoid circular at module level
-            result = await env.step(NL2SQLAction(query=sql))
+            result = await env.step({"query": sql}) # changed
             obs = result.observation
 
             reward = obs.reward or 0.0

server/app.py

@@ -30,6 +30,11 @@ app = create_fastapi_app(
     observation_cls=NL2SQLObservation
 )
 
+@app.on_event("startup")
+async def startup_event():
+    from db.seed import seed_database
+    seed_database() # This creates the 'ecommerce.db' file inside the container
+
 def main():
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)