Out of Memory

I am using a NANO, which whilst i know is not the most powerful machine in the world but i keep getting out of memory errors whilst going through tutorials.

following this one: Transfer learning with TensorFlow Hub  |  TensorFlow Core

and I get OOM when allocating tensor with shape[32,3,224,224] at the prediction stage.

I have had loads of errors like that when trying to make my own models, which I kinda understood as I was making layers but I have a preloaded one here?

Jetpack 4.4.1, tensorflow 2.3.1+nv20.10