Spaces:

ccmusic-database
/

bel_canto

Running

MuGeminorum commited on Mar 26

Commit

e2817c2

•

1 Parent(s): 4c6af99

fix img1k

Files changed (1) hide show

model.py CHANGED Viewed

@@ -18,7 +18,7 @@ def get_backbone(ver, backbone_list):
 def model_info(m_ver):
     backbone_list = load_dataset(
         "monet-joe/cv_backbones",
-        split="ImageNet1k_v1",
         trust_remote_code=True,
     )
     backbone = get_backbone(m_ver, backbone_list)

 def model_info(m_ver):
     backbone_list = load_dataset(
         "monet-joe/cv_backbones",
+        split="IMAGENET1K_V1",
         trust_remote_code=True,
     )
     backbone = get_backbone(m_ver, backbone_list)