change train method

Only build the model once when the TextPro is inited