Update app.py

app.py

@@ -1,9 +1,21 @@
 import tensorflow as tf
+from keras.utils import custom_object_scope
 import gradio as gr
 from PIL import Image, ImageDraw
 
-# Carregue o modelo
-loaded_model = tf.keras.models.load_model('modelo_treinado.h5')
+# Defina a camada personalizada FixedDropout
+class FixedDropout(tf.keras.layers.Dropout):
+    def __init__(self, rate, **kwargs):
+        super().__init__(rate, **kwargs)
+        self._rate = rate
+
+    def call(self, inputs):
+        return tf.nn.dropout(inputs, self._rate)
+
+# Registre a camada personalizada FixedDropout com o TensorFlow
+with custom_object_scope({'FixedDropout': FixedDropout}):
+    # Carregue o modelo
+    loaded_model = tf.keras.models.load_model('modelo_treinado.h5')
 
 # Crie uma lista de classes
 class_names = ["Normal", "Cataract"]