ariG23498 HF Staff commited on
Commit
35301df
·
1 Parent(s): d2ff88f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,12 +19,12 @@ initialize(config_path="configs", version_base=None)
19
  from huggingface_hub import Repository
20
 
21
  repo = Repository(
22
- local_dir="models",
23
  clone_from="ariG23498/clip-dinoiser",
24
  use_auth_token=os.environ.get("token")
25
  )
26
 
27
- check_path = 'models/checkpoints/last.pt'
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
  check = torch.load(check_path, map_location=device)
 
19
  from huggingface_hub import Repository
20
 
21
  repo = Repository(
22
+ local_dir="clip-dinoiser",
23
  clone_from="ariG23498/clip-dinoiser",
24
  use_auth_token=os.environ.get("token")
25
  )
26
 
27
+ check_path = 'clip-dinoiser/checkpoints/last.pt'
28
  device = "cuda" if torch.cuda.is_available() else "cpu"
29
 
30
  check = torch.load(check_path, map_location=device)