Update app.py

app.py

@@ -11,16 +11,13 @@ from tensorflow.keras.models import Model
 
 
 # load vgg16 model
-pre_trained_model = EfficientNetB7(weights='imagenet', include_top=False, input_shape=(224, 224, 3))
+pre_trained = EfficientNetB7(weights='imagenet', include_top=False, input_shape=(224, 224, 3))
 # Freeze the base model
-pre_trained_model.trainable = False
+pre_trained.trainable = False
 
-pre_trained_model = tf.keras.Sequential([
-    pre_trained_model,
-    tf.keras.layers.GlobalAveragePooling2D()
-])
+x = tf.keras.layers.GlobalAveragePooling2D()(pre_trained.output)
 # restructure the model
-pre_trained_model = Model(inputs=pre_trained_model.inputs, outputs=pre_trained_model.layers[-1].output)
+pre_trained_model = Model(inputs=pre_trained.input, outputs=x)
 
 model = tf.keras.models.load_model("Image_Captioner_model.h5")