Adding GATED_IMAGENET token for authentication

app.py

@@ -1,3 +1,4 @@
+import os
 import gradio as gr
 import random
 import time
@@ -15,6 +16,7 @@ https://huggingface.co/datasets/datacomp
 NUM_EXAMPLES = 1281167
 # Arbitrary small number, only using in devv'ing (uncomment #.take(DEV_AMOUNT) below to use it).
 DEV_AMOUNT = 100
+GATED_IMAGENET = os.environ.get("GATED_IMAGENET")
 
 
 def main(percentage=10):
@@ -30,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).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.