Update app.py

app.py

@@ -47,8 +47,18 @@ def load_image_model():
 #     encoder=tf.keras.models.load_model('./encoder_model.h5')
 #     return encoder
 
+# InceptionResNetV2 takes (299, 299, 3) image as inputs
+# and return features in (8, 8, 1536) shape
+FEATURE_EXTRACTOR = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(
+    include_top=False, weights="imagenet"
+)
+
 # **** ENCODER ****
 image_input = Input(shape=(IMG_HEIGHT, IMG_WIDTH, IMG_CHANNELS))
+image_features = FEATURE_EXTRACTOR(image_input)
+x = Reshape((FEATURES_SHAPE[0] * FEATURES_SHAPE[1], FEATURES_SHAPE[2]))(
+    image_features
+)
 encoder_output = Dense(ATTENTION_DIM, activation="relu")(x)
 encoder = tf.keras.Model(inputs=image_input, outputs=encoder_output)
 # **** ENCODER ****