diff --git a/demos/image_classification_with_string_output/download_model.py b/demos/image_classification_with_string_output/download_model.py
index 1056d1930e..7e361c32e5 100644
--- a/demos/image_classification_with_string_output/download_model.py
+++ b/demos/image_classification_with_string_output/download_model.py
@@ -57,7 +57,7 @@ def serving_fn(image):
 
     return serving_fn
 
-model = keras.applications.MobileNet()
+model = keras.applications.MobileNetV2()
 model_dir = "./model"
 model_version = 1
 model_export_path = f"{model_dir}/{model_version}"