changed layer

app.py

@@ -7,9 +7,10 @@ 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
+# 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[-2].output)
+mobilenet_model = Model(inputs=mobilenet_model.inputs, outputs=mobilenet_model.layers[-1].output)
+
 
 # Load the trained captioning model
 model = tf.keras.models.load_model("model_9.h5")