Spaces:

Rolv-Arild
/

next-goal-predictor

Running

Rolv-Arild commited on Jul 2

Commit

d33fd9c

verified ·

1 Parent(s): 81966e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,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-daily-energy.pt", map_location=DEVICE)
 MODEL.eval()