okeowo1014 commited on
Commit
2b3b6cd
1 Parent(s): c6cc42e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,9 +3,12 @@ import cv2
3
  import numpy as np
4
  from tensorflow.keras.preprocessing import image
5
  import tensorflow as tf
 
 
 
6
 
7
  # Load the saved model (replace with your model filename)
8
- model = tf.keras.models.load_model('cat_dog_classifier.keras')
9
 
10
  # Image dimensions for the model
11
  img_width, img_height = 224, 224
 
3
  import numpy as np
4
  from tensorflow.keras.preprocessing import image
5
  import tensorflow as tf
6
+ from huggingface_hub import from_pretrained_keras
7
+
8
+ model = from_pretrained_keras("okeowo1014/catsanddogs")
9
 
10
  # Load the saved model (replace with your model filename)
11
+ # model = tf.keras.models.load_model('cat_dog_classifier.keras')
12
 
13
  # Image dimensions for the model
14
  img_width, img_height = 224, 224