Thanh Vinh Vo commited on
Commit
4e45f5c
·
1 Parent(s): a9be9cc
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -40,7 +40,7 @@ def get_file_location(file_name: str) -> str:
40
  class BasicAgent:
41
  def __init__(self):
42
  print("BasicAgent initialized.")
43
- self.multimodal_agent = CodeAgent(
44
  tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
45
  model=InferenceClientModel(
46
  model_id="meta-llama/Llama-3.2-11B-Vision-Instruct",
@@ -63,7 +63,7 @@ class BasicAgent:
63
  description="I can understand natural language and I am optimised for visual recognition, image reasoning, captioning, and answering general questions about an image.",
64
  verbosity_level=0,
65
  max_steps=10,
66
- )
67
 
68
  self.code_agent = CodeAgent(
69
  tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
@@ -95,7 +95,9 @@ class BasicAgent:
95
  "Qwen/Qwen2.5-32B-Instruct"
96
  ),
97
  tools=[get_file_location],
98
- managed_agents=[self.multimodal_agent, self.code_agent],
 
 
99
  additional_authorized_imports=[
100
  "requests",
101
  "bs4",
 
40
  class BasicAgent:
41
  def __init__(self):
42
  print("BasicAgent initialized.")
43
+ """ self.multimodal_agent = CodeAgent(
44
  tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
45
  model=InferenceClientModel(
46
  model_id="meta-llama/Llama-3.2-11B-Vision-Instruct",
 
63
  description="I can understand natural language and I am optimised for visual recognition, image reasoning, captioning, and answering general questions about an image.",
64
  verbosity_level=0,
65
  max_steps=10,
66
+ ) """
67
 
68
  self.code_agent = CodeAgent(
69
  tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
 
95
  "Qwen/Qwen2.5-32B-Instruct"
96
  ),
97
  tools=[get_file_location],
98
+ managed_agents=[
99
+ # self.multimodal_agent,
100
+ self.code_agent],
101
  additional_authorized_imports=[
102
  "requests",
103
  "bs4",