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)