Clementina Tom (via Gemini) commited on
Commit
e5cd6dd
·
1 Parent(s): a30026f

Fix model paths: looking in models/ folder

Browse files
Files changed (1) hide show
  1. plrs/model/model_loader.py +3 -3
plrs/model/model_loader.py CHANGED
@@ -41,9 +41,9 @@ def load_model_from_hub(device: str = "cpu"):
41
  return None, "huggingface_hub not installed"
42
 
43
  for filename, model_type in [
44
- ("sakt_decay_best.pt", "SAKTWithDecay"),
45
- ("sakt_vanilla_best.pt", "SAKTModel"),
46
- ("sakt_model.pt", "SAKTModel"),
47
  ]:
48
  try:
49
  path = hf_hub_download(repo_id=HF_REPO, filename=filename)
 
41
  return None, "huggingface_hub not installed"
42
 
43
  for filename, model_type in [
44
+ ("models/sakt_decay_best.pt", "SAKTWithDecay"),
45
+ ("models/sakt_vanilla_best.pt", "SAKTModel"),
46
+ ("models/sakt_model.pt", "SAKTModel"),
47
  ]:
48
  try:
49
  path = hf_hub_download(repo_id=HF_REPO, filename=filename)