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:
|