mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Simplify gallery

Browse source
This commit is contained in:
oobabooga 2023-04-05 00:34:17 -03:00
parent 4ab679480e
commit ca8bb38949
1 changed files with 1 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
extensions/gallery/script.py
View file

@ -66,13 +66,7 @@ def generate_html():
container_html = '<div class="character-container">'
image_html = "<div class='placeholder'></div>"
for i in [
f"characters/{character}.png",
f"characters/{character}.jpg",
f"characters/{character}.jpeg",
]:
path = Path(i)
for path in [Path(f"characters/{character}.{extension}") for extension in ['png', 'jpg', 'jpeg']]:
if path.exists():
image_html = f'<img src="file/{get_image_cache(path)}">'
break