Created by: Steffy-zxf
As the demo text-classification use the GPU to run the model, it isn't suitable. We fix it that user can choose the gpu or not to run the model.