Update app.py

app.py

@@ -27,7 +27,13 @@ def preprocess(ex):
     return ex
 
 # Apply preprocessing
-ds = ds.map(preprocess, remove_columns=["image", "label"])
+ds = ds.map(
+    preprocess,
+    remove_columns=["image", "label"],
+    num_proc=1,
+    load_from_cache_file=False
+)
+
 
 # Model config
 model.config.decoder_start_token_id = processor.tokenizer.cls_token_id