mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-11-22 08:07:56 +01:00
Include the bot's image as base64
This is needed for Colab.
This commit is contained in:
parent
cb226247e8
commit
c4e87c109e
@ -4,6 +4,7 @@ This is a library for formatting GPT-4chan and chat outputs as nice HTML.
|
|||||||
|
|
||||||
'''
|
'''
|
||||||
|
|
||||||
|
import base64
|
||||||
import copy
|
import copy
|
||||||
import re
|
import re
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@ -260,7 +261,12 @@ def generate_chat_html(history, name1, name2, character):
|
|||||||
]:
|
]:
|
||||||
|
|
||||||
if Path(i).exists():
|
if Path(i).exists():
|
||||||
img = f'<img src="file/{i}">'
|
with open(i, "rb") as image_file:
|
||||||
|
encoded_string = base64.b64encode(image_file.read())
|
||||||
|
if i.endswith('png'):
|
||||||
|
img = f'<img src="data:image/png;base64,{encoded_string.decode("utf-8")}">'
|
||||||
|
elif i.endswith('jpg') or i.endswith('jpeg'):
|
||||||
|
img = f'<img src="data:image/jpg;base64,{encoded_string.decode("utf-8")}">'
|
||||||
break
|
break
|
||||||
|
|
||||||
img_me = ''
|
img_me = ''
|
||||||
|
Loading…
Reference in New Issue
Block a user