I'm not sure if the GPU is powerful enough on the RPi, maybe something like a Jetson Nano could handle running ML models better.