Update README.md
Browse files
README.md
CHANGED
@@ -24,7 +24,7 @@ from huggingface_hub import hf_hub_download
|
|
24 |
CKPT_SAVE_PATH = [your checkpoint storage path]
|
25 |
CKPT_NAME = [checkpoint name]
|
26 |
hf_hub_download(local_dir=CKPT_SAVE_PATH,
|
27 |
-
repo_id='
|
28 |
filename=CKPT_NAME)
|
29 |
model = models_vit.__dict__['vit_base_patch16'](
|
30 |
num_classes=2,
|
|
|
24 |
CKPT_SAVE_PATH = [your checkpoint storage path]
|
25 |
CKPT_NAME = [checkpoint name]
|
26 |
hf_hub_download(local_dir=CKPT_SAVE_PATH,
|
27 |
+
repo_id='Wolowolo/fsfm-3c',
|
28 |
filename=CKPT_NAME)
|
29 |
model = models_vit.__dict__['vit_base_patch16'](
|
30 |
num_classes=2,
|