Spaces:
Running
Running
check
Browse files
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="
|
23 |
clone_from="ariG23498/clip-dinoiser",
|
24 |
use_auth_token=os.environ.get("token")
|
25 |
)
|
26 |
|
27 |
-
check_path = '
|
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)
|