ZennyKenny commited on
Commit
9f75d2d
·
verified ·
1 Parent(s): 96def24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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=HfApiModel(model_id="meta-llama/Llama-3.3-70B-Instruct"),
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: