Spaces:
Runtime error
Runtime error
Commit
•
a46df70
1
Parent(s):
2a20d87
check if None added
Browse files- app.py +3 -6
- images/banner.jpg +0 -0
- models_config.yaml +2 -1
app.py
CHANGED
@@ -35,14 +35,11 @@ face_cascade = cv2.CascadeClassifier(config['FD_M_PATH'])
|
|
35 |
|
36 |
def main_pipeline(image):
|
37 |
|
38 |
-
|
39 |
-
|
|
|
40 |
|
41 |
-
#image = im_b64.file.read()
|
42 |
|
43 |
-
#image = Image.open(image)
|
44 |
-
|
45 |
-
#image = image.convert("RGB")
|
46 |
original_image = image
|
47 |
image = np.asarray(image)
|
48 |
|
|
|
35 |
|
36 |
def main_pipeline(image):
|
37 |
|
38 |
+
if image is None:
|
39 |
+
image = Image.open(config['BANNER_IMG'])
|
40 |
+
return image, "NO FACE DETECTED", "NO FACE DETECTED"
|
41 |
|
|
|
42 |
|
|
|
|
|
|
|
43 |
original_image = image
|
44 |
image = np.asarray(image)
|
45 |
|
images/banner.jpg
ADDED
![]() |
models_config.yaml
CHANGED
@@ -2,4 +2,5 @@ A_M_PATH: models;age_model.h5
|
|
2 |
G_M_PATH: models;gender_model.h5
|
3 |
FD_M_PATH: models;haarcascade_frontalface_default.xml
|
4 |
|
5 |
-
TEST_IMG_PATH: images
|
|
|
|
2 |
G_M_PATH: models;gender_model.h5
|
3 |
FD_M_PATH: models;haarcascade_frontalface_default.xml
|
4 |
|
5 |
+
TEST_IMG_PATH: images
|
6 |
+
BANNER_IMG: images;banner.jpg
|