ABVM commited on
Commit
b4c715a
·
verified ·
1 Parent(s): 391ac5c

Update OpenRouter_Agent.py

Browse files
Files changed (1) hide show
  1. OpenRouter_Agent.py +6 -6
OpenRouter_Agent.py CHANGED
@@ -27,11 +27,11 @@ HF_API_TOKEN = os.getenv("HF_API_TOKEN")
27
 
28
  #)
29
 
30
- object_detection_tool = Tool.from_space(
31
- space_id = "stevengrove/YOLO-World",
32
- name = "Real-Time Open-Vocabulary Object Detector",
33
- description = "Detect objects in images or videos."
34
- )
35
 
36
 
37
  OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
@@ -90,7 +90,7 @@ class MultiAgentSystem:
90
 
91
  self.info_agent = CodeAgent(
92
  model =self.qwen_model,
93
- tools=[PythonInterpreterTool(), object_detection_tool ],
94
  name="info_agent",
95
  description=(
96
  "You are an agent tasked with cleaning, parsing, calculating information, and performing OCR if images are provided in the {task}. "
 
27
 
28
  #)
29
 
30
+ #object_detection_tool = Tool.from_space(
31
+ # space_id = "stevengrove/YOLO-World",
32
+ # name = "Real-Time Open-Vocabulary Object Detector",
33
+ # description = "Detect objects in images or videos."
34
+ #)
35
 
36
 
37
  OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
 
90
 
91
  self.info_agent = CodeAgent(
92
  model =self.qwen_model,
93
+ tools=[PythonInterpreterTool()],
94
  name="info_agent",
95
  description=(
96
  "You are an agent tasked with cleaning, parsing, calculating information, and performing OCR if images are provided in the {task}. "