Spaces:

bgaspra
/

CivitAI

Runtime error

bgaspra commited on Aug 1, 2024

Commit

7a1f71f

verified ·

1 Parent(s): 597527d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ import matplotlib.pyplot as plt
 dataset = load_dataset("thefcraft/civitai-stable-diffusion-337k")
 # Take a subset of the dataset
-subset_size = 10000
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 # Directory to save images

 dataset = load_dataset("thefcraft/civitai-stable-diffusion-337k")
 # Take a subset of the dataset
+subset_size = 5000
 dataset_subset = dataset['train'].shuffle(seed=42).select(range(subset_size))
 # Directory to save images