Separated the model creation from the 'text' generation.