Going to process full dataset.

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)#.take(DEV_AMOUNT)
 
     # Create a set of indices that are randomly chosen, to change their labels.
     # Specifically, randomly choose NUM_EXAMPLES/percentage indices.