def infer(self, data):
    output_data = self.model.predict(data)
    pred = {'label': output_data}
    return pred
def infer(self, data):
    output_data = self.model.predict(data)
    pred = {'label': output_data}
    return pred