Update app.py

app.py

@@ -47,6 +47,12 @@ def load_image_model():
 #     encoder=tf.keras.models.load_model('./encoder_model.h5')
 #     return encoder
 
+# **** ENCODER ****
+image_input = Input(shape=(IMG_HEIGHT, IMG_WIDTH, IMG_CHANNELS))
+encoder_output = Dense(ATTENTION_DIM, activation="relu")(x)
+encoder = tf.keras.Model(inputs=image_input, outputs=encoder_output)
+# **** ENCODER ****
+
 
 # **** DECODER ****
 
@@ -90,11 +96,6 @@ decoder_pred_model = tf.keras.Model(
 )
 # **** DECODER ****
 
-# **** ENCODER ****
-image_input = Input(shape=(IMG_HEIGHT, IMG_WIDTH, IMG_CHANNELS))
-encoder_output = Dense(ATTENTION_DIM, activation="relu")(x)
-encoder = tf.keras.Model(inputs=image_input, outputs=encoder_output)
-# **** ENCODER ****
     
 st.title(":blue[Nishant Guvvada's] :red[AI Journey]  Image Caption Generation")
 image = Image.open('./title.jpg')