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 = 1000
+subset_size = 50
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 
 # Directory to save images