Upload app.py

app.py

@@ -12,7 +12,7 @@ from transformers import (
 )
 from datasets import load_dataset
 
-wikiart_dataset = load_dataset("huggan/wikiart", split="train")
+wikiart_dataset = load_dataset("huggan/wikiart", split="train", streaming=True)
 device = torch.device("cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu")
 
 blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning-base")