Spaces:
Running
on
Zero
Running
on
Zero
Update demo.py
Browse files
demo.py
CHANGED
@@ -115,11 +115,7 @@ def process_image(input_image):
|
|
115 |
gaze_points = []
|
116 |
|
117 |
for face in faces:
|
118 |
-
|
119 |
-
(face["x_min"] + face["x_max"]) / 2,
|
120 |
-
(face["y_min"] + face["y_max"]) / 2,
|
121 |
-
)
|
122 |
-
gaze = model.detect_gaze(pil_image, face_center)["gaze"]
|
123 |
|
124 |
if gaze is None:
|
125 |
continue
|
|
|
115 |
gaze_points = []
|
116 |
|
117 |
for face in faces:
|
118 |
+
gaze = model.detect_gaze(pil_image, face=face)["gaze"]
|
|
|
|
|
|
|
|
|
119 |
|
120 |
if gaze is None:
|
121 |
continue
|