iarbel
/

yolov8s_test

Object Detection

ultralyticsplus

Model card Files Files and versions

iarbel commited on Jan 4, 2024

Commit

5007afa

·

1 Parent(s): 3a43c10

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -33,7 +33,7 @@ class EndpointHandler():
         self.model.overrides['max_det'] = config.get('max_det')
         # Get label idx-to-name
-        names = model.model.names
         # perform inference
         result = self.model.predict(inputs['image'])[0]

         self.model.overrides['max_det'] = config.get('max_det')
         # Get label idx-to-name
+        names = self.model.model.names
         # perform inference
         result = self.model.predict(inputs['image'])[0]