sudip2003 commited on
Commit
d15307f
·
verified ·
1 Parent(s): e95b46a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from tensorflow.keras.preprocessing import image
4
  import numpy as np
5
 
6
  # Load the trained model
7
- model = tf.keras.models.load_model('cat_dog_classifier_vgg16.h5')
8
 
9
  # Define a function to make predictions
10
  def predict_image(img):
 
4
  import numpy as np
5
 
6
  # Load the trained model
7
+ model = tf.keras.models.load_model('Model_catsVSdogs.h5')
8
 
9
  # Define a function to make predictions
10
  def predict_image(img):