Update app.py

app.py

@@ -32,7 +32,7 @@ def main(percentage=10):
     # Load the dataset from the HF hub. Use streaming so as not to load the entire dataset at once.
     # Use the .take(DEV_AMOUNT) to only grab a small chunk of instances to develop with.
     dataset = load_dataset("ILSVRC/imagenet-1k", split="train", streaming=True,
-                           trust_remote_code=True, token=GATED_IMAGENET)#.take(DEV_AMOUNT)
+                           trust_remote_code=True, token=GATED_IMAGENET, columns=["label"]).take(DEV_AMOUNT)
 
     # Create a set of indices that are randomly chosen, to change their labels.
     # Specifically, randomly choose NUM_EXAMPLES/percentage indices.
@@ -45,7 +45,7 @@ def main(percentage=10):
 
     # Upload the new version of the dataset (this will take awhile)
     Dataset.from_generator(updated_dataset.__iter__).push_to_hub(
-        "datacomp/imagenet-1k-random" + str(percentage), token=GATED_IMAGENET)
+        "datacomp/imagenet-1k-random-debug" + str(percentage), token=GATED_IMAGENET)
 
     end = time.time()
     print("That took %d seconds" % (end - start))