Update app.py

app.py

@@ -41,10 +41,10 @@ def load_decoder_model():
     decoder_model=tf.keras.models.load_model('./decoder_pred_model.h5')
     return decoder_model
 
-@st.cache_resource()
-def load_encoder_model():
-    encoder=tf.keras.models.load_model('./encoder_model.h5')
-    return encoder
+# @st.cache_resource()
+# def load_encoder_model():
+#     encoder=tf.keras.models.load_model('./encoder_model.h5')
+#     return encoder
 
 
 FEATURE_EXTRACTOR = tf.keras.applications.inception_resnet_v2.InceptionResNetV2(include_top=False, weights="imagenet")
@@ -96,7 +96,7 @@ def predict_caption(file):
     resize = tf.image.resize(image, (IMG_HEIGHT, IMG_WIDTH))
     img = resize/255
     
-    encoder = load_encoder_model()
+    # encoder = load_encoder_model()
     features = encoder(tf.expand_dims(img, axis=0))
     dec_input = tf.expand_dims([word_to_index("<start>")], 1)
     result = []