theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-23 00:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update 1_Image_Creator.py

Browse Source
This commit is contained in:
Artiprocher
2024-01-04 15:22:36 +08:00
committed by GitHub
parent 759f5e33eb
commit 32c28ef4b6
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pages/1_Image_Creator.py
View File

@@ -6,6 +6,7 @@ st.set_page_config(layout="wide")
from streamlit_drawable_canvas import st_canvas from streamlit_drawable_canvas import st_canvas
from diffsynth.models import ModelManager from diffsynth.models import ModelManager
from diffsynth.pipelines import SDImagePipeline, SDXLImagePipeline from diffsynth.pipelines import SDImagePipeline, SDXLImagePipeline
from diffsynth.data.video import crop_and_resize
config = { config = {
@@ -176,7 +177,7 @@ with column_input:
upload_image = st.file_uploader("Upload image", type=["png", "jpg"], key="upload_image") upload_image = st.file_uploader("Upload image", type=["png", "jpg"], key="upload_image")
if upload_image is not None: if upload_image is not None:
st.session_state["input_image"] = Image.open(upload_image) st.session_state["input_image"] = crop_and_resize(Image.open(upload_image), height, width)
elif create_white_board: elif create_white_board:
st.session_state["input_image"] = Image.fromarray(np.ones((height, width, 3), dtype=np.uint8) * 255) st.session_state["input_image"] = Image.fromarray(np.ones((height, width, 3), dtype=np.uint8) * 255)
else: else: