Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -102,9 +102,13 @@ def handle_query(user_input: str) -> str:
|
|
102 |
"""
|
103 |
return query_sql(user_input)
|
104 |
|
|
|
|
|
|
|
|
|
105 |
agent = CodeAgent(
|
106 |
tools=[sql_engine],
|
107 |
-
model=
|
108 |
)
|
109 |
|
110 |
with gr.Blocks() as demo:
|
|
|
102 |
"""
|
103 |
return query_sql(user_input)
|
104 |
|
105 |
+
model = HfApiModel(
|
106 |
+
model_id="meta-llama/Llama-3.3-70B-Instruct",
|
107 |
+
token=os.getenv("HF_API_TOKEN")
|
108 |
+
)
|
109 |
agent = CodeAgent(
|
110 |
tools=[sql_engine],
|
111 |
+
model=model
|
112 |
)
|
113 |
|
114 |
with gr.Blocks() as demo:
|