VanShingel commited on
Commit
a46df70
1 Parent(s): 2a20d87

check if None added

Browse files
Files changed (3) hide show
  1. app.py +3 -6
  2. images/banner.jpg +0 -0
  3. 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
- #file = request.form()
39
- #im_b64 = file['img']
 
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