Update app.py

app.py

@@ -48,7 +48,13 @@ def load_decoder_model():
 #     return encoder
 
 
-FEATURE_EXTRACTOR = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(include_top=False, weights="imagenet")
+@st.cache_resource()
+def feature_extractor():
+    FEATURE_EXTRACTOR=tf.keras.models.load_model('./feature_extractor.h5')
+    return FEATURE_EXTRACTOR
+
+
+FEATURE_EXTRACTOR = feature_extractor()
 FEATURE_EXTRACTOR.trainable = False
 image_input = Input(shape=(IMG_HEIGHT, IMG_WIDTH, IMG_CHANNELS))
 image_features = FEATURE_EXTRACTOR(image_input)