Update app.py

app.py

@@ -36,7 +36,7 @@ GATED_IMAGENET = os.environ.get("GATED_IMAGENET")
 
 
 def create_subset_dataset(dataset, fraction_size):
-    dataset = dataset.shuffle(buffer_size=NUM_EXAMPLES)
+    dataset = dataset.shuffle()
     num_samples = int(NUM_EXAMPLES / fraction_size)
     sampled_dataset = dataset.take(num_samples)   
     return sampled_dataset