5m4ck3r commited on
Commit
c50f5c0
1 Parent(s): a7ce092

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -47,9 +47,11 @@ def processFaceDetails(image):
47
  if len(data) > 1:
48
  cordinates = data[1]["box"]
49
  crop_box = (data[1]["box"]['xmin'], data[1]["box"]['ymin'], data[1]["box"]['xmax'], data[1]["box"]['ymax'])
50
- else:
51
  cordinates = data[0]["box"]
52
  crop_box = (data[0]["box"]['xmin'], data[0]["box"]['ymin'], data[0]["box"]['xmax'], data[0]["box"]['ymax'])
 
 
53
  cropped_image = image.crop(crop_box)
54
  facedata = faceModal(cropped_image)
55
  for imask in facedata:
 
47
  if len(data) > 1:
48
  cordinates = data[1]["box"]
49
  crop_box = (data[1]["box"]['xmin'], data[1]["box"]['ymin'], data[1]["box"]['xmax'], data[1]["box"]['ymax'])
50
+ elif len(data) > 0:
51
  cordinates = data[0]["box"]
52
  crop_box = (data[0]["box"]['xmin'], data[0]["box"]['ymin'], data[0]["box"]['xmax'], data[0]["box"]['ymax'])
53
+ else:
54
+ return ret
55
  cropped_image = image.crop(crop_box)
56
  facedata = faceModal(cropped_image)
57
  for imask in facedata: