ElenaRyumina
commited on
Commit
β’
e761828
1
Parent(s):
1cb6819
Update app/app_utils.py
Browse files- 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 |
-
|
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 |
|