Update app.py

app.py

@@ -52,7 +52,7 @@ def main(percentage=10):
         dataset = load_dataset("datacomp/imagenet-1k-random" + str(percentage), split="train", streaming=True,
                            trust_remote_code=True, token=GATED_IMAGENET).take(DEV_AMOUNT)
     else:
-        dataset = load_dataset("datacomp/imagenet-1k-random" + str(percentage), split="train", streaming=True,
+        dataset = load_dataset("datacomp/imagenet-1k-random" + str(percentage) + "-take2", split="train", streaming=True,
                            trust_remote_code=True, token=GATED_IMAGENET)
 
     for frac in FRACTIONS:
@@ -64,7 +64,7 @@ def main(percentage=10):
                 "datacomp/debug-imagenet-1k-random-" + str(percentage) + '-frac-1over' + str(frac), token=GATED_IMAGENET)
         else:
             Dataset.from_generator(sampled_dataset.__iter__).push_to_hub(
-                "datacomp/imagenet-1k-random-" + str(percentage) + '-frac-1over' + str(frac), token=GATED_IMAGENET)
+                "datacomp/imagenet-1k-random-" + str(percentage) + '-frac-1over' + str(frac) + "-take2", token=GATED_IMAGENET)
 
 
     end = time.time()