Rupesx007 commited on
Commit
e9707b0
·
verified ·
1 Parent(s): 97f1555

edited app.py (activated the model)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import io
6
 
7
  # Load your trained model
8
  custom_objects = {'BatchNormalization': tf.keras.layers.BatchNormalization}
9
- # model = tf.keras.models.load_model('ResNet152V2.h5')
10
 
11
 
12
  # Define class labels of the animals
 
6
 
7
  # Load your trained model
8
  custom_objects = {'BatchNormalization': tf.keras.layers.BatchNormalization}
9
+ model = tf.keras.models.load_model('ResNet152V2.h5')
10
 
11
 
12
  # Define class labels of the animals