def METHOD_NAME(self, data):
    # Generate predicted sentences
    word_idx = mx.nd.argmax(data[0], axis=1).asnumpy()
    res = ""
    for idx in word_idx:
        res += self.idx2word[idx] + " "
    ret = {"prediction": res}
    return [ret]