Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|