Using a retrained TensorFlow model for TensorFlow Lite in iOS