Spaces:
Running
Running
Update app.py
Browse files
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-
|
| 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 |
|