Spaces:
Sleeping
Sleeping
Daniel kiani
commited on
fix: Updated path in app.py
Browse files- 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=
|
| 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")
|