Check the examples TensorFlow Lite provides to see what suits your case best.
  
      

      TensorFlow
  

  
    

TensorFlow Lite Examples | Machine Learning Mobile Apps 1

  Sample ML apps for Android, iOS and Raspberry Pi. See end-to-end examples with complete instructions to train, test and deploy models on mobile devices.


  

  
    
    
  

  


If you need any help ping me.
Regards