Samka1u commited on
Commit
330c4f0
·
verified ·
1 Parent(s): 4325cbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -205,7 +205,7 @@ def agent_setup():
205
  Args:
206
  None
207
  Returns:
208
- sql_agent (Agent): The agent that will be used for inference.
209
  """
210
  sql_model = InferenceClientModel(
211
  api_key=NEBIUS_API_KEY,
 
205
  Args:
206
  None
207
  Returns:
208
+ sql_agent (CodeAgent): The agent that will be used for inference.
209
  """
210
  sql_model = InferenceClientModel(
211
  api_key=NEBIUS_API_KEY,