Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 from sqlalchemy import create_engine, MetaData, Table, Column, String, Integer, Float, insert, text, inspect
 from smolagents import tool, CodeAgent, InferenceClientModel
@@ -80,9 +81,13 @@ for table in ["receipts", "waiters"]:
 sql_engine.description = updated_description
 
 # --- Create the agent ---
+# Use the HF_TOKEN secret stored in the Space
 agent = CodeAgent(
     tools=[sql_engine],
-    model=InferenceClientModel("meta-llama/Meta-Llama-3-8B-Instruct"),
+    model=InferenceClientModel(
+        "meta-llama/Meta-Llama-3-8B-Instruct",
+        api_key=os.environ.get("HF_TOKEN")  # 👈 Use secret key
+    ),
 )
 
 # --- Define Gradio interface ---