From 01514cd3d19fc2d48feca47d4f5e285c8ed5a2bc Mon Sep 17 00:00:00 2001 From: Symbiomatrix Date: Fri, 12 Jan 2024 04:14:48 +0200 Subject: [PATCH] Update previews.py --- scripts/previews.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/previews.py b/scripts/previews.py index 60bf532..c752aa8 100644 --- a/scripts/previews.py +++ b/scripts/previews.py @@ -48,8 +48,9 @@ def load_previews(): if hash.lower() not in hashes: continue images = r['images'] if (nsfw_previews is False): images = [i for i in images if i['nsfw'] is False] - if (len(images) == 0): continue - image_url = images[0]['url'] + image_urls = [image['url'] for image in images if (image['type'] == 'image')] + if (len(image_urls) == 0): continue + image_url = image_urls[0] civitai.update_resource_preview(hash, image_url) updated += 1