Spaces:
Running
on
Zero
Running
on
Zero
Update demo.py
Browse files
demo.py
CHANGED
|
@@ -16,7 +16,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 16 |
trust_remote_code=True,
|
| 17 |
torch_dtype=torch.float16,
|
| 18 |
device_map={"": "cuda"},
|
| 19 |
-
revision="
|
| 20 |
)
|
| 21 |
|
| 22 |
def visualize_faces_and_gaze(face_boxes, gaze_points=None, image=None, show_plot=True):
|
|
|
|
| 16 |
trust_remote_code=True,
|
| 17 |
torch_dtype=torch.float16,
|
| 18 |
device_map={"": "cuda"},
|
| 19 |
+
revision="64ce9d45e8603058826ab339d3bb0f48f7289b1f"
|
| 20 |
)
|
| 21 |
|
| 22 |
def visualize_faces_and_gaze(face_boxes, gaze_points=None, image=None, show_plot=True):
|