Daniel kiani commited on
Commit
c6233c8
·
verified ·
1 Parent(s): f91b841

fix: Updated path in app.py

Browse files
Files changed (1) hide show
  1. scripts/app.py +1 -1
scripts/app.py CHANGED
@@ -27,7 +27,7 @@ def load_artifacts():
27
  print("--- Loading all artifacts for the Gradio app ---")
28
 
29
  # HF-FRIENDLY: Path is relative, assuming the checkpoint is in the root of the Space repo.
30
- CHECKPOINT_PATH = "sasrec-epoch=05-val_hitrate@10=0.3614.ckpt"
31
  DATA_FOLDER = "data/"
32
 
33
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
27
  print("--- Loading all artifacts for the Gradio app ---")
28
 
29
  # HF-FRIENDLY: Path is relative, assuming the checkpoint is in the root of the Space repo.
30
+ CHECKPOINT_PATH = "checkpoints/sasrec-epoch=06-val_hitrate@10=0.3629.ckpt"
31
  DATA_FOLDER = "data/"
32
 
33
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")