def METHOD_NAME(self, input_ids):
    """
    Predict the class  for a review / sentence whether
    it is belong to world / sports / business /sci-tech
    :param encoding: Input encoding to be passed through the layers for prediction
    :return: output - predicted output
    """
    inputs = self.input_ids.to(self.device)
    self.outputs = self.model.forward(inputs)
    self.out = np.argmax(self.outputs.cpu().detach())
    return [self.out.item()]