Hi, welcome to the forum.
What sgugger said was pretty much what you should do to achieve what you are looking for. But, if you are still not clear, you can refer to this thread, “How do we use our model against a specific image?”. This was discussed at length there together with Jeremy’s replies.
It’s a bit tricky to get this all to works. Fret not, if you are looking for the latest codes to get this working, here’s the direct link to my reply in the same thread. I hope it helps in some ways.