Update app.py

app.py

@@ -2,6 +2,19 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 
+# Define a custom layer for FixedDropout
+class FixedDropout(tf.keras.layers.Dropout):
+    def _get_noise_shape(self, inputs):
+        if self.noise_shape is None:
+            return self.noise_shape
+        symbolic_shape = tf.shape(inputs)
+        noise_shape = [symbolic_shape[axis] if shape is None else shape
+                       for axis, shape in enumerate(self.noise_shape)]
+        return tuple(noise_shape)
+
+# Register the custom layer
+tf.keras.utils.get_custom_objects()['FixedDropout'] = FixedDropout
+
 # Load your trained TensorFlow model
 model = tf.keras.models.load_model('modelo_treinado.h5')  # Load your saved model