Manith Marapperuma
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,12 @@ import io
|
|
8 |
|
9 |
# Load the model (ensure correct path for loading)
|
10 |
# For Hugging Face Spaces, ensure the model file is included in the repository or loaded from a URL
|
11 |
-
|
|
|
|
|
|
|
|
|
|
|
12 |
|
13 |
# Load the MTCNN face detection model
|
14 |
mtcnn = MTCNN()
|
|
|
8 |
|
9 |
# Load the model (ensure correct path for loading)
|
10 |
# For Hugging Face Spaces, ensure the model file is included in the repository or loaded from a URL
|
11 |
+
|
12 |
+
|
13 |
+
input_shape = (224, 224, 3)
|
14 |
+
model = tf.keras.models.load_model('oily_dry.h5', compile=False, custom_objects={'input_shape': input_shape})
|
15 |
+
|
16 |
+
# model = tf.keras.models.load_model('oily_dry.h5')
|
17 |
|
18 |
# Load the MTCNN face detection model
|
19 |
mtcnn = MTCNN()
|