Spaces:

bfshi
/

AutoGaze

Running on Zero

bfshi commited on Mar 10

Commit

4d4269e

1 Parent(s): 9053eb9

update model

Files changed (1) hide show

demo_utils.py CHANGED Viewed

@@ -52,7 +52,7 @@ def image_to_video(image_path, output_path, fps):
 def load_model(device='cuda'):
     print("Loading AutoGaze model from HuggingFace...")
-    model = AutoGaze.from_pretrained("bfshi/AutoGaze")
     model = model.to(device)
     model.eval()

 def load_model(device='cuda'):
     print("Loading AutoGaze model from HuggingFace...")
+    model = AutoGaze.from_pretrained("nvidia/AutoGaze")
     model = model.to(device)
     model.eval()