Hugging Face's logo

Spaces:
soiz1
/
Gaze-LLE
Runtime error

App Files Community
soiz1 commited on
Commit
cdf137d
·
verified ·
1 Parent(s): 5586578

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ from retinaface import RetinaFace
27
  device = "cuda" if torch.cuda.is_available() else "cpu"
28
 
29
  # load Gaze-LLE model
30
- model, transform = torch.hub.load("fkryan/gazelle", "gazelle_dinov2_vitl14_inout", use_memory_efficient_attention=False)
31
  model.eval()
32
  model.to(device)
33
 
 
27
  device = "cuda" if torch.cuda.is_available() else "cpu"
28
 
29
  # load Gaze-LLE model
30
+ model, transform = torch.hub.load("fkryan/gazelle", "gazelle_dinov2_vitl14_inout")
31
  model.eval()
32
  model.to(device)
33