Spaces:

Rolv-Arild
/

next-goal-predictor

Running

Rolv-Arild commited on Oct 1

Commit

4ff5b1f

verified ·

1 Parent(s): 74f9c53

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