Update app.py

app.py

@@ -4,14 +4,14 @@ import requests
 import cv2
 import numpy as np
 
-# Define a custom layer 'FixedDropout'
+# Define a custom layer 'FixedDropout' without specifying 'name' argument
 def fixed_dropout(x):
     return tf.keras.layers.Dropout(0.5)(x)
 
 # Load the TensorFlow model while registering the custom layer
-tf.keras.utils.get_custom_objects()['FixedDropout'] = fixed_dropout
+custom_objects = {'fixed_dropout': fixed_dropout}
 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 = tf.keras.models.load_model(tf_model_path, custom_objects=custom_objects)
 
 class_labels = ["Normal", "Cataract"]
 
@@ -31,8 +31,7 @@ def predict(inp):
 demo = gr.Interface(
     fn=predict,
     inputs=gr.inputs.Image(type="pil"),
-    outputs=["label", "number"],
-    examples=[["cheetah.jpg"]],
+    outputs=["label", "number"]
 )
 
 demo.launch()