JuyeopDang commited on
Commit
e1cbb0b
·
verified ·
1 Parent(s): 81c9ab8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ HF_TOKEN = os.environ['HF_TOKEN']
15
  class BasicAgent:
16
  def __init__(self):
17
  print("BasicAgent initialized.")
18
- self.agent = InferenceClientModel(tools=[DuckDuckGoSearchTool()])
19
 
20
  def __call__(self, question: str) -> str:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
15
  class BasicAgent:
16
  def __init__(self):
17
  print("BasicAgent initialized.")
18
+ self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel())
19
 
20
  def __call__(self, question: str) -> str:
21
  print(f"Agent received question (first 50 chars): {question[:50]}...")