michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

inspiration perfected

Browse source
This commit is contained in:
yfszzx 2022-10-22 10:21:21 +08:00
parent 40ddb6df61
commit d93ea5cdeb
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
modules/inspiration.py
View file

@ -73,8 +73,11 @@ def get_inspiration_images(source, types, keyword):
image_list = []
for a in names:
image_path = os.path.join(opts.inspiration_dir, a)
images = os.listdir(image_path)
image_list.append((os.path.join(image_path, random.choice(images)), a))
images = os.listdir(image_path)
if len(images) > 0:
image_list.append((os.path.join(image_path, random.choice(images)), a))
else:
print(image_path)
return image_list, names
def select_click(index, name_list):