Update app.py

app.py

@@ -8,6 +8,8 @@ from tensorflow.keras.preprocessing.image import load_img, img_to_array
 from tensorflow.keras.preprocessing.text import Tokenizer
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.models import Model
+from tensorflow.keras.models import model_from_json
+from keras.optimizers import Adam
 
 
 # load vgg16 model
@@ -21,6 +23,22 @@ pre_trained_model = Model(inputs=pre_trained.input, outputs=x)
 
 # model = tf.keras.models.load_model("image_captioning_30k_model.h5")
 
+###########################################################################################################
+# Load model architecture
+with open("30k_model_architecture.json", "r") as json_file:
+    loaded_model_json = json_file.read()
+    
+# Redefine the optimizer
+optimizer = Adam(lr=0.001)
+
+# Load weights
+model = model_from_json(loaded_model_json)
+model.load_weights("30k_model_weights.h5")
+
+# Load optimizer state
+model.compile(optimizer=optimizer, loss='categorical_crossentropy')
+###########################################################################################################
+
 tokenizer = Tokenizer()
 with open("Image_Captioner_tokenizer_30k.pkl", "rb") as f:
     tokenizer = pickle.load(f)