Hi what you are doing is correct but I will not recommend this method. Kindly have a look at the following example and see how how its implemented inside a model.
Also you can start with Deep Learning Onramp which is free and will give you all basics you need to know for building and testing and applying models.
