Update app.py

app.py

@@ -18,21 +18,21 @@ model = Xception()
 # Restructure model
 model = Model(inputs = model.inputs , outputs = model.layers[-2].output)
 
-import tensorflow as tf
+# 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
+# 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
+#     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})
+# # 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: