Update models.py

models.py

@@ -535,11 +535,11 @@ class ImageCaptioningModel(keras.Model):
         # called automatically.
         return [self.loss_tracker, self.acc_tracker]
 
-cnn_model = get_cnn_model()
-encoder = TransformerEncoderBlock(embed_dim=EMBED_DIM, dense_dim=FF_DIM, num_heads=1)
-decoder = TransformerDecoderBlock(embed_dim=EMBED_DIM, ff_dim=FF_DIM, num_heads=2)
-new_model = ImageCaptioningModel(
-    cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation,
-)
-
-new_model.load_weights('model_weights.h5')
+# cnn_model = get_cnn_model()
+# encoder = TransformerEncoderBlock(embed_dim=EMBED_DIM, dense_dim=FF_DIM, num_heads=1)
+# decoder = TransformerDecoderBlock(embed_dim=EMBED_DIM, ff_dim=FF_DIM, num_heads=2)
+# new_model = ImageCaptioningModel(
+#     cnn_model=cnn_model, encoder=encoder, decoder=decoder, image_aug=image_augmentation,
+# )
+
+# new_model.load_weights('model_weights.h5')