def inference()

in deployment/handler.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def inference(self, img):
        """
        Predict the chip stack mask of an image using a trained deep learning model.
        """
        logger.info(f"Device on inference is: {self.device}")
        self.model.eval()
        inputs = Variable(img).to(self.device)
        outputs = self.model.forward(inputs)
        logging.debug(outputs.shape)
        return outputs