Update app.py

app.py

@@ -3,9 +3,23 @@ import tensorflow as tf
 import cv2
 import numpy as np
 
-# Load the TensorFlow model
+# Define the custom layer 'FixedDropout'
+class FixedDropout(tf.keras.layers.Layer):
+    def __init__(self, rate, **kwargs):
+        super(FixedDropout, self).__init__(**kwargs)
+        self.rate = rate
+
+    def call(self, inputs, training=None):
+        return tf.keras.backend.dropout(inputs, level=self.rate)
+
+# Load the TensorFlow model with custom layer handling
+def load_model_with_custom_objects(model_path):
+    with tf.keras.utils.custom_object_scope({'FixedDropout': FixedDropout}):
+        model = tf.keras.models.load_model(model_path)
+    return model
+
 tf_model_path = 'modelo_treinado.h5'  # Update with the path to your TensorFlow model
-tf_model = tf.keras.models.load_model(tf_model_path)
+tf_model = load_model_with_custom_objects(tf_model_path)
 
 class_labels = ["Normal", "Cataract"]