changed layer

app.py

@@ -7,9 +7,9 @@ from tensorflow.keras.models import Model
 from tensorflow.keras.applications.mobilenet_v2 import MobileNetV2, preprocess_input
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 
-# Load MobileNetV2 model for feature extraction (using the final layer)
-mobilenet_model = MobileNetV2(weights="imagenet")
-mobilenet_model = Model(inputs=mobilenet_model.inputs, outputs=mobilenet_model.layers[-1].output)
+# Load MobileNetV2 model for feature extraction (with pooling and no top layer)
+mobilenet_model = MobileNetV2(weights="imagenet", include_top=False, pooling='avg')
+mobilenet_model = Model(inputs=mobilenet_model.inputs, outputs=mobilenet_model.output)
 
 
 # Load the trained captioning model