Ubik80 commited on
Commit
a0a32e9
·
verified ·
1 Parent(s): a13608d

added new tool

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -4,17 +4,18 @@ 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
- from tools import search_tool, weather_info_tool
8
 
9
 
10
  # Setup agent
11
  model = HfApiModel()
12
  alfred = CodeAgent(
13
- tools=[guest_info_tool, search_tool, weather_info_tool],
14
  model=model
15
  )
16
 
17
 
 
18
  # Gradio interface function
19
  def ask_alfred(question):
20
  return alfred.run(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, weather_info_tool, hub_stats_tool
8
 
9
 
10
  # Setup agent
11
  model = HfApiModel()
12
  alfred = CodeAgent(
13
+ tools=[guest_info_tool, search_tool, weather_info_tool, hub_stats_tool],
14
  model=model
15
  )
16
 
17
 
18
+
19
  # Gradio interface function
20
  def ask_alfred(question):
21
  return alfred.run(question)