huytofu92 commited on
Commit
e789e42
·
1 Parent(s): 29cd08b

Fix tool desc last

Browse files
Files changed (1) hide show
  1. vlm_tools.py +2 -2
vlm_tools.py CHANGED
@@ -133,7 +133,7 @@ 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
  }
@@ -167,7 +167,7 @@ class OCRTool(Tool):
167
  description = "Scan an image for text. It takes a list of frames (images) as input and returns a list of text in the images."
168
  name = "ocr_scan"
169
  inputs = {
170
- "frames": {"type": "List[str]", "description": "The list of frames (images) to scan for text"}
171
  }
172
  output_type = "List[List[str]]"
173
 
 
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": "any", "description": "The list of frames (images) to detect objects in. Must be a List[str] or a List[np.ndarray]"},
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
  }
 
167
  description = "Scan an image for text. It takes a list of frames (images) as input and returns a list of text in the images."
168
  name = "ocr_scan"
169
  inputs = {
170
+ "frames": {"type": "any", "description": "The list of frames (images) to scan for text. Must be a List[str] or a List[np.ndarray]"}
171
  }
172
  output_type = "List[List[str]]"
173