Spaces:

moondream
/

gaze-demo

Running on Zero

vikhyatk commited on Jan 6

Commit

6be4105

verified ·

1 Parent(s): c5f0574

Update demo.py

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -18,7 +18,7 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True,
     torch_dtype=torch.float16,
     device_map={"": "cuda"},
-    revision="84db61d0ff2dca21475a11a3c0e3080e2d2723d5"
 )
@@ -119,7 +119,7 @@ def process_image(input_image):
                 (face["x_min"] + face["x_max"]) / 2,
                 (face["y_min"] + face["y_max"]) / 2,
             )
-            gaze = model.detect_gaze(enc_image, face_center)
             if gaze is None:
                 continue

     trust_remote_code=True,
     torch_dtype=torch.float16,
     device_map={"": "cuda"},
+    revision="6357578ba3bf29aad59900d22751102d175f4976"
 )
                 (face["x_min"] + face["x_max"]) / 2,
                 (face["y_min"] + face["y_max"]) / 2,
             )
+            gaze = model.detect_gaze(enc_image, face_center)["gaze"]
             if gaze is None:
                 continue