Sygil-Dev/sygil-webui
1
1
Fork 0
mirror of https://github.com/Sygil-Dev/sygil-webui.git synced 2024-12-14 22:13:41 +03:00
Code Issues Projects Releases Wiki Activity

added auto conversion for 4 channel PNG to RGB

Browse Source
This commit is contained in:
Hafiidz 2022-09-07 14:29:49 +08:00
parent 00d36bfabb
commit 432ebdad29
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/webui_streamlit.py
View File

@ -1537,7 +1537,7 @@ def layout():
with preview_tab: with preview_tab:
#st.write("Image") #st.write("Image")
#Image for testing #Image for testing
#image = Image.open(requests.get("https://icon-library.com/images/image-placeholder-icon/image-placeholder-icon-13.jpg", stream=True).raw) #image = Image.open(requests.get("https://icon-library.com/images/image-placeholder-icon/image-placeholder-icon-13.jpg", stream=True).raw).convert('RGB')
#new_image = image.resize((175, 240)) #new_image = image.resize((175, 240))
#preview_image = st.image(image) #preview_image = st.image(image)
@ -1691,7 +1691,7 @@ def layout():
st.session_state["editor_image"] = editor_image st.session_state["editor_image"] = editor_image
if uploaded_images: if uploaded_images:
image = Image.open(uploaded_images) image = Image.open(uploaded_images).convert('RGB')
#img_array = np.array(image) # if you want to pass it to OpenCV #img_array = np.array(image) # if you want to pass it to OpenCV
new_img = image.resize((width, height)) new_img = image.resize((width, height))
st.image(new_img) st.image(new_img)
@ -1713,7 +1713,7 @@ def layout():
message = st.empty() message = st.empty()
#if uploaded_images: #if uploaded_images:
#image = Image.open(uploaded_images) #image = Image.open(uploaded_images).convert('RGB')
##img_array = np.array(image) # if you want to pass it to OpenCV ##img_array = np.array(image) # if you want to pass it to OpenCV
#new_img = image.resize((width, height)) #new_img = image.resize((width, height))
#st.image(new_img, use_column_width=True) #st.image(new_img, use_column_width=True)
@ -1724,7 +1724,7 @@ def layout():
# load the models when we hit the generate button for the first time, it wont be loaded after that so dont worry. # load the models when we hit the generate button for the first time, it wont be loaded after that so dont worry.
load_models(False, use_GFPGAN, use_RealESRGAN, RealESRGAN_model) load_models(False, use_GFPGAN, use_RealESRGAN, RealESRGAN_model)
if uploaded_images: if uploaded_images:
image = Image.open(uploaded_images) image = Image.open(uploaded_images).convert('RGB')
new_img = image.resize((width, height)) new_img = image.resize((width, height))
#img_array = np.array(image) # if you want to pass it to OpenCV #img_array = np.array(image) # if you want to pass it to OpenCV