Device-based Models with TensorFlow Lite