Update app.py

app.py

@@ -18,7 +18,22 @@ model = Xception()
 # Restructure model
 model = Model(inputs = model.inputs , outputs = model.layers[-2].output)
 
-caption_model = tf.keras.models.load_model('model.h5')
+import tensorflow as tf
+
+class MyEmbedding(tf.keras.layers.Embedding):
+    def __init__(self, *args, input_length=None, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.input_length = input_length
+
+    def from_config(cls, config):
+        input_length = config.pop('input_length', None)
+        instance = cls(**config)
+        instance.input_length = input_length
+        return instance
+
+# Load the model with custom objects
+caption_model = tf.keras.models.load_model('model.h5', custom_objects={'MyEmbedding': MyEmbedding})
+
 
 with open('captions.txt', 'r') as f:
     next(f)