Thanh Vinh Vo commited on
Commit
2e0c033
·
1 Parent(s): 4e45f5c
Files changed (1) hide show
  1. app.py +7 -6
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",
@@ -60,11 +60,12 @@ class BasicAgent:
60
  "bytes",
61
  ],
62
  name="multimodal_agent",
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],
70
  model=InferenceClientModel(
@@ -94,9 +95,9 @@ class BasicAgent:
94
  model=InferenceClientModel(
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",
 
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",
 
60
  "bytes",
61
  ],
62
  name="multimodal_agent",
63
+ description="""
64
+ I can understand natural language and I am optimised for visual recognition, image reasoning, captioning, and answering general questions about an image.""",
65
  verbosity_level=0,
66
  max_steps=10,
67
+ )
68
+
69
  self.code_agent = CodeAgent(
70
  tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
71
  model=InferenceClientModel(
 
95
  model=InferenceClientModel(
96
  "Qwen/Qwen2.5-32B-Instruct"
97
  ),
98
+ tools=[VisitWebpageTool(), DuckDuckGoSearchTool(), get_file_location],
99
  managed_agents=[
100
+ self.multimodal_agent,
101
  self.code_agent],
102
  additional_authorized_imports=[
103
  "requests",