ElenaRyumina commited on
Commit
e761828
β€’
1 Parent(s): 1cb6819

Update app/app_utils.py

Browse files
Files changed (1) hide show
  1. app/app_utils.py +4 -2
app/app_utils.py CHANGED
@@ -57,8 +57,10 @@ def preprocess_image_and_predict(inp):
57
  cur_face_hm = np.float32(cur_face_hm) / 255
58
  heatmap = show_cam_on_image(cur_face_hm, grayscale_cam, use_rgb=True)
59
 
60
- return cur_face, heatmap, confidences
61
-
 
 
62
 
63
  def preprocess_video_and_predict(video):
64
 
 
57
  cur_face_hm = np.float32(cur_face_hm) / 255
58
  heatmap = show_cam_on_image(cur_face_hm, grayscale_cam, use_rgb=True)
59
 
60
+ return cur_face, heatmap, confidences
61
+
62
+ else:
63
+ return None, None
64
 
65
  def preprocess_video_and_predict(video):
66