Solved: Plot function to visualize image predictions

Please close this thread. I found my solution. In the end, I used excel to demo a few predictions after attempts to get matplotlib to work.

Original post:
I’m trying to visualize my predictions the same way as this photo

Is anyone aware of any handy package to do this kind of plots?