Prediction

[harirajeev]

Thanks for your code.
Could you please add a section for prediction using the trained weights ?

[marzieh-abdolmaleki]

hi
can you please add prediction part?