Update app.py

app.py

@@ -4,7 +4,12 @@ import requests
 import cv2
 import numpy as np
 
-# Load the TensorFlow model
+# Define a custom layer 'FixedDropout'
+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
 tf_model_path = 'modelo_treinado.h5'  # Update with the path to your TensorFlow model
 tf_model = tf.keras.models.load_model(tf_model_path)
 
@@ -26,7 +31,8 @@ def predict(inp):
 demo = gr.Interface(
     fn=predict,
     inputs=gr.inputs.Image(type="pil"),
-    outputs=["label", "number"]
+    outputs=["label", "number"],
+    examples=[["cheetah.jpg"]],
 )
 
 demo.launch()