Manith Marapperuma commited on
Commit
9f3c439
·
verified ·
1 Parent(s): ec6413d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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
- model = tf.keras.models.load_model('oily_dry.h5')
 
 
 
 
 
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()