I was working on an image recognition problem. After training the model, I saved the architecture as well as weights. Now I want to use the model for extracting features from ot