Spaces:
Running
Running
MuGeminorum
commited on
Commit
•
e8ff109
1
Parent(s):
5295064
fix img1k
Browse files
model.py
CHANGED
@@ -18,7 +18,7 @@ def get_backbone(ver, backbone_list):
|
|
18 |
def model_info(m_ver):
|
19 |
backbone_list = load_dataset(
|
20 |
"monet-joe/cv_backbones",
|
21 |
-
split="
|
22 |
trust_remote_code=True,
|
23 |
)
|
24 |
backbone = get_backbone(m_ver, backbone_list)
|
|
|
18 |
def model_info(m_ver):
|
19 |
backbone_list = load_dataset(
|
20 |
"monet-joe/cv_backbones",
|
21 |
+
split="IMAGENET1K_V1",
|
22 |
trust_remote_code=True,
|
23 |
)
|
24 |
backbone = get_backbone(m_ver, backbone_list)
|