Rolv-Arild commited on
Commit
bcb8f9a
·
verified ·
1 Parent(s): 36647e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ from rlgym_tools.rocket_league.replays.convert import replay_to_rlgym
13
  from rlgym_tools.rocket_league.replays.parsed_replay import ParsedReplay
14
  from tqdm import trange, tqdm
15
 
16
- os.chmod("/usr/local/lib/python3.10/site-packages/rlgym_tools/rocket_league/replays/carball", 0o755)
17
 
18
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
 
@@ -21,7 +21,7 @@ repo = Repository(local_dir="vortex-ngp", clone_from="Rolv-Arild/vortex-ngp", to
21
  repo.git_pull()
22
 
23
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
- MODEL = torch.jit.load("vortex-ngp/vortex-ngp-rosy-shape.pt", map_location=DEVICE)
25
  MODEL.eval()
26
 
27
 
 
13
  from rlgym_tools.rocket_league.replays.parsed_replay import ParsedReplay
14
  from tqdm import trange, tqdm
15
 
16
+ # os.chmod("/usr/local/lib/python3.10/site-packages/rlgym_tools/rocket_league/replays/carball", 0o755)
17
 
18
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
19
 
 
21
  repo.git_pull()
22
 
23
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
24
+ MODEL = torch.jit.load("vortex-ngp/vortex-ngp-glowing-dragon.pt", map_location=DEVICE)
25
  MODEL.eval()
26
 
27