Thank you so much for your reply , I am now able to load the weights thanks to you.
One extra question, I did exactly what you told me and this is what I got:

image1379×605 36.8 KB

I want to use this model to make a prediction for one image.
Could you please help me figure out how to do so?