Skip to content

Commit 1c44173

Browse files
add __model__ to detection
Signed-off-by: Ashwin Vaidya <[email protected]>
1 parent e809f4e commit 1c44173

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

model_api/python/model_api/models/detection_model.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ class DetectionModel(ImageModel):
2828
The `postprocess` method must be implemented in a specific inherited wrapper.
2929
"""
3030

31+
__model__ = "DetectionModel"
32+
3133
def __init__(self, inference_adapter, configuration=dict(), preload=False):
3234
"""Detection Model constructor
3335

0 commit comments

Comments
 (0)