Spaces:
Sleeping
Sleeping
Fixing input size
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ import cv2
|
|
5 |
import tensorflow as tf
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
-
IMG_HEIGHT =
|
9 |
-
IMG_WIDTH =
|
10 |
|
11 |
# Load the saved Keras model
|
12 |
model = load_model("model_01.keras")
|
|
|
5 |
import tensorflow as tf
|
6 |
from tensorflow.keras.models import load_model
|
7 |
|
8 |
+
IMG_HEIGHT = 96
|
9 |
+
IMG_WIDTH = 96
|
10 |
|
11 |
# Load the saved Keras model
|
12 |
model = load_model("model_01.keras")
|