Update handler.py

handler.py

@@ -50,7 +50,7 @@ class EndpointHandler():
             A :obj:`list` | `dict`: will be serialized and returned
         """
         # Get the prediction
-        result = self.model(data['inputs'])
+        result = self.yolov8_model(data['inputs'])
         # Get the original image with channel shifted
         img = result[0].orig_img[:,:,::-1]
         H, W, _ = img.shape