Ubik80 commited on
Commit
2c8aea7
·
verified ·
1 Parent(s): 469630e

added search tool

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,10 +4,12 @@ os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")
4
  import gradio as gr
5
  from smolagents import CodeAgent, HfApiModel
6
  from retriever import guest_info_tool
 
7
 
8
  # Setup agent
9
  model = HfApiModel()
10
- alfred = CodeAgent(tools=[guest_info_tool], model=model)
 
11
 
12
  # Gradio interface function
13
  def ask_alfred(question):
 
4
  import gradio as gr
5
  from smolagents import CodeAgent, HfApiModel
6
  from retriever import guest_info_tool
7
+ from tools import search_tool
8
 
9
  # Setup agent
10
  model = HfApiModel()
11
+ alfred = CodeAgent(tools=[guest_info_tool, search_tool], model=model)
12
+
13
 
14
  # Gradio interface function
15
  def ask_alfred(question):