at1300 commited on
Commit
af205c8
·
verified ·
1 Parent(s): fb8f05d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -6,7 +6,7 @@ import pandas as pd
6
  import yaml
7
 
8
  from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool, VisitWebpageTool
9
- from tools import visit_webpage, transcribe_audio, analyze_video
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
@@ -24,7 +24,8 @@ class coder_agent:
24
  tools=[
25
  #visit_webpage,
26
  VisitWebpageTool(),
27
- WebSearchTool()
 
28
  ],
29
  name="web_agent",
30
  description="Browses the web to find information",
@@ -70,7 +71,9 @@ class MasterAgent:
70
 
71
  self.master_agent = CodeAgent(
72
  tools=[VisitWebpageTool(),
73
- WebSearchTool()],
 
 
74
  model=model,
75
  add_base_tools=True,
76
  planning_interval=5,
 
6
  import yaml
7
 
8
  from smolagents import CodeAgent, WebSearchTool, InferenceClientModel, DuckDuckGoSearchTool, Tool, VisitWebpageTool
9
+ from tools import visit_webpage, analyze_image # transcribe_audio, analyze_video
10
 
11
  # (Keep Constants as is)
12
  # --- Constants ---
 
24
  tools=[
25
  #visit_webpage,
26
  VisitWebpageTool(),
27
+ WebSearchTool(),
28
+ analyze_image
29
  ],
30
  name="web_agent",
31
  description="Browses the web to find information",
 
71
 
72
  self.master_agent = CodeAgent(
73
  tools=[VisitWebpageTool(),
74
+ WebSearchTool(),
75
+ analyze_image,
76
+ ],
77
  model=model,
78
  add_base_tools=True,
79
  planning_interval=5,