Hi Rahul,
you can also install pytorch directly on Android using Termux/Archlinux
then take picture with Termux-API [termux-camera-photo]
The performance is enough for testings https://youtu.be/ox9TxZhmJ30
I started this github to describe how to do it.
Regards