Spaces:

Rolv-Arild
/

next-goal-predictor

Sleeping

Rolv-Arild commited on Jun 29

Commit

f95a08d

verified ·

1 Parent(s): 355e78e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ repo = Repository(local_dir="vortex-ngp", clone_from="Rolv-Arild/vortex-ngp", to
 repo.git_pull()
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-MODEL = torch.jit.load("vortex-ngp/vortex_ngp-avid-frog.pt", map_location=DEVICE)
 MODEL.eval()

 repo.git_pull()
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+MODEL = torch.jit.load("vortex-ngp/vortex-ngp-avid-frog.pt", map_location=DEVICE)
 MODEL.eval()