Spaces:
Sleeping
Sleeping
Fix tool
Browse files- 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
|