Update app.py

app.py

@@ -29,6 +29,7 @@ FRACTIONS = [2, 4, 8, 16, 32, 64]
 DEV_AMOUNT = 10
 if DEV:
     NUM_EXAMPLES = DEV_AMOUNT
+    FRACTIONS = [2]
 # Whether to read in the distribution over labels from an external text file.
 READ_DISTRO = False
 GATED_IMAGENET = os.environ.get("GATED_IMAGENET")
@@ -59,10 +60,10 @@ def main(percentage=10):
 
         # Upload the new version of the dataset (this will take awhile)
         if DEV:
-            Dataset.from_generator(updated_dataset.__iter__).push_to_hub(
+            Dataset.from_generator(sampled_dataset.__iter__).push_to_hub(
                 "datacomp/debug-imagenet-1k-random-" + "-" + str(percentage) + '-frac-1/' + str(frac), token=GATED_IMAGENET)
         else:
-            Dataset.from_generator(updated_dataset.__iter__).push_to_hub(
+            Dataset.from_generator(sampled_dataset.__iter__).push_to_hub(
                 "datacomp/imagenet-1k-random" + str(percentage) + '-frac-1/' + str(frac), token=GATED_IMAGENET)