Update app.py

app.py

@@ -2,16 +2,7 @@ import numpy as np
 import tensorflow as tf
 from tensorflow.keras.layers import BatchNormalization
 
-model = tf.keras.models.load_model('4 - animals_mobilenetV3_1_new.h5')
-
-# 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)
+model = tf.keras.models.load_model('best_model.h5')
 
 labels = ['Clams', 'Corals', 'Crabs', 'Dolphin', 'Eel', 'Fish',
               'Jelly Fish', 'Lobster', 'Nudibranchs', 'Octopus', 'Otter',