Spaces:

Rolv-Arild
/

next-goal-predictor

Sleeping

Rolv-Arild commited on Jun 29

Commit

f79e46f

verified ·

1 Parent(s): 09e6bb0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 from huggingface_hub import Repository
-repo = Repository(local_dir="vortex-ngp", clone_from="Rolv-Arild/vortex-ngp")
 repo.git_pull()
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 from huggingface_hub import Repository
+repo = Repository(local_dir="vortex-ngp", clone_from="Rolv-Arild/vortex-ngp", token=os.getenv("HF_TOKEN"))
 repo.git_pull()
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")