Update main.py

main.py

@@ -149,8 +149,10 @@ Now you'll create a tf.keras model where the output layer is the last convolutio
 
 """
 
-image_model = tf.keras.applications.InceptionV3(include_top=False,
-                                                weights='imagenet')
+# image_model = tf.keras.applications.InceptionV3(include_top=False,
+#                                                 weights='imagenet') 
+
+image_model = tf.keras.models.load_model("incep.h5")
 new_input = image_model.input
 hidden_layer = image_model.layers[-1].output