huytofu92 commited on
Commit
d6c2e86
·
1 Parent(s): b28ae74
Files changed (1) hide show
  1. vlm_tools.py +2 -2
vlm_tools.py CHANGED
@@ -130,14 +130,14 @@ onnx_path = "vlm_assets/yolov3-8.onnx"
130
  names_path = "vlm_assets/obj.names"
131
 
132
  class ObjectDetectionTool(Tool):
133
- description = "Detect objects in a list of frames (images)"
134
  name = "object_detection"
135
  inputs = {
136
  "frames": {"type": List[str], "description": "The list of frames (images) to detect objects in"},
137
  "onnx_path": {"type": "string", "description": "The path to the onnx file"},
138
  "names_path": {"type": "string", "description": "The path to the names file"}
139
  }
140
- output_type = List[List[str]]
141
 
142
  def setup(self):
143
  # Load ONNX model
 
130
  names_path = "vlm_assets/obj.names"
131
 
132
  class ObjectDetectionTool(Tool):
133
+ description = "Detect objects in a list of frames (images). It takes a list of frames (images) as input and returns a list of detected objects with labels, confidence, and bounding boxes."
134
  name = "object_detection"
135
  inputs = {
136
  "frames": {"type": List[str], "description": "The list of frames (images) to detect objects in"},
137
  "onnx_path": {"type": "string", "description": "The path to the onnx file"},
138
  "names_path": {"type": "string", "description": "The path to the names file"}
139
  }
140
+ output_type = "List[List[str]]"
141
 
142
  def setup(self):
143
  # Load ONNX model