Update app.py

app.py

@@ -10,7 +10,7 @@ class FixedDropout(tf.keras.layers.Layer):
         self.rate = rate
 
     def call(self, inputs, training=None):
-        return tf.keras.backend.dropout(inputs, level=self.rate)
+        return tf.keras.layers.Dropout(self.rate)(inputs, training=training)
 
 # Load the TensorFlow model with custom layer handling
 def load_model_with_custom_objects(model_path):
@@ -41,4 +41,4 @@ def classify_image(input_image):
 input_image = gr.inputs.Image(shape=(224, 224, 3))  # Define the input image shape
 output_label = gr.outputs.Label()  # Define the output label
 
-gr.Interface(fn=classify_image, inputs=input_image, outputs=output_label).launch()
+gr.Interface(fn=classify_image, inputs=input_image, outputs=output_label).launch()