Ubik80 commited on
Commit
b41134d
·
verified ·
1 Parent(s): bbdf2e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,11 +4,15 @@ 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
 
8
 
9
  # Setup agent
10
  model = HfApiModel()
11
- alfred = CodeAgent(tools=[guest_info_tool, search_tool], model=model)
 
 
 
12
 
13
 
14
  # Gradio interface function
 
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