From 051728617aa85bce03cdc7685951d3b8b0b04a09 Mon Sep 17 00:00:00 2001
From: Tirumalesh <111861663+tiruk007@users.noreply.github.com>
Date: Tue, 20 Dec 2022 18:50:21 +0530
Subject: [PATCH] Update __init__.py

Update the example code
---
 tensorflow_examples/lite/model_maker/public/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tensorflow_examples/lite/model_maker/public/__init__.py b/tensorflow_examples/lite/model_maker/public/__init__.py
index 827637c347c..1c0f4246a60 100644
--- a/tensorflow_examples/lite/model_maker/public/__init__.py
+++ b/tensorflow_examples/lite/model_maker/public/__init__.py
@@ -23,6 +23,9 @@
 Typical usage of Model Maker is to create a model in a few lines of code, e.g.:
 
 ```python
+from tflite_model_maker import image_classifier
+from tflite_model_maker.image_classifier import DataLoader
+
 # Load input data specific to an on-device ML app.
 data = DataLoader.from_folder('flower_photos/')
 train_data, test_data = data.split(0.9)