meg HF Staff commited on
Commit
1e07ef1
·
verified ·
1 Parent(s): 3d85131

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def main(percentage=10):
49
  dataset = load_dataset("ILSVRC/imagenet-1k", split="train", streaming=True,
50
  trust_remote_code=True, token=GATED_IMAGENET)
51
 
52
- label_list, label_fractions = get_label_fractions()
53
 
54
  # How many new random labels are we creating?
55
  num_new_labels = int(round(NUM_EXAMPLES/float(percentage)))
 
49
  dataset = load_dataset("ILSVRC/imagenet-1k", split="train", streaming=True,
50
  trust_remote_code=True, token=GATED_IMAGENET)
51
 
52
+ label_list, label_fractions = get_label_fractions(dataset)
53
 
54
  # How many new random labels are we creating?
55
  num_new_labels = int(round(NUM_EXAMPLES/float(percentage)))