Update app.py

app.py

@@ -14,7 +14,7 @@ import gradio as gr
 dataset = load_dataset("thefcraft/civitai-stable-diffusion-337k")
 
 # Take a subset of the dataset
-subset_size = 50  # Reduce the subset size for faster execution in Spaces
+subset_size = 5000  # Reduce the subset size for faster execution in Spaces
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images