Spaces:

datacomp
/

select-subset

Sleeping

meg HF Staff commited on Sep 27, 2024

Commit

1e07ef1

verified ·

1 Parent(s): 3d85131

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -49,7 +49,7 @@ def main(percentage=10):
         dataset = load_dataset("ILSVRC/imagenet-1k", split="train", streaming=True,
                            trust_remote_code=True, token=GATED_IMAGENET)
-    label_list, label_fractions = get_label_fractions()
     # How many new random labels are we creating?
     num_new_labels = int(round(NUM_EXAMPLES/float(percentage)))

         dataset = load_dataset("ILSVRC/imagenet-1k", split="train", streaming=True,
                            trust_remote_code=True, token=GATED_IMAGENET)
+    label_list, label_fractions = get_label_fractions(dataset)
     # How many new random labels are we creating?
     num_new_labels = int(round(NUM_EXAMPLES/float(percentage)))