Optimizing the output of classification probablity in demo\text_classification\predict.py and corresponding README.md