Update app.py

app.py

@@ -1,10 +1,17 @@
 import numpy as np
 import tensorflow as tf
+from tensorflow.keras.layers import BatchNormalization
 
 model = tf.keras.models.load_model('4 - animals_mobilenetV3_1.h5')
 
-import numpy as np
-import tensorflow as tf
+# Function to remove '/' from layer names
+def remove_slash(name):
+    return name.replace('/', '_')
+
+# Modify layer names
+for layer in model.layers:
+    if isinstance(layer, BatchNormalization):
+        layer.name = remove_slash(layer.name)
 
 labels = ['Clams', 'Corals', 'Crabs', 'Dolphin', 'Eel', 'Fish',
               'Jelly Fish', 'Lobster', 'Nudibranchs', 'Octopus', 'Otter',