Update app.py

app.py

@@ -3,8 +3,9 @@ import gradio as gr
 import random
 import time
 from datasets import load_dataset, Dataset
-from PIL import PngImagePlugin
-PngImagePlugin.MAX_TEXT_CHUNK = 1048576 * 10 # this is the current value
+from PIL import PngImagePlugin, ImageFile
+ImageFile.LOAD_TRUNCATED_IMAGES = True
+PngImagePlugin.MAX_TEXT_CHUNK = 1048576 * 10 # this is 10x the amount.
 """
 This code is designed to read in the ImageNet 1K ILSVRC dataset from the Hugging Face Hub, 
 then create a new version of this dataset with {percentage} lines with random labels between 0-9,