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

Allow jpg profile images

Browse source
This commit is contained in:
oobabooga 2023-01-15 15:45:25 -03:00
parent ada2e556d0
commit 027c3dd27d
3 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
html_generator.py
View file

@ -216,6 +216,10 @@ def generate_chat_html(history, name1, name2):
output += f'<style>{css}</style><div class="chat" id="chat">'
if Path("profile.png").exists():
img = '<img src="file/profile.png">'
elif Path("profile.jpg").exists():
img = '<img src="file/profile.jpg">'
elif Path("profile.jpeg").exists():
img = '<img src="file/profile.jpeg">'
else:
img = ''