vikhyatk commited on
Commit
1161baa
·
verified ·
1 Parent(s): 6be4105

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -119,7 +119,7 @@ def process_image(input_image):
119
  (face["x_min"] + face["x_max"]) / 2,
120
  (face["y_min"] + face["y_max"]) / 2,
121
  )
122
- gaze = model.detect_gaze(enc_image, face_center)["gaze"]
123
 
124
  if gaze is None:
125
  continue
 
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