theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-24 18:28:10 +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:44:31 +08:00
committed by GitHub
parent 32c28ef4b6
commit 49db42bf1f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pages/1_Image_Creator.py
View File

@@ -61,7 +61,7 @@ def load_model(model_type, model_path):
return model_manager, pipeline return model_manager, pipeline
def use_output_image_as_input(): def use_output_image_as_input(update=True):
# Search for input image # Search for input image
output_image_id = 0 output_image_id = 0
selected_output_image = None selected_output_image = None
@@ -72,8 +72,9 @@ def use_output_image_as_input():
selected_output_image = st.session_state["output_images"][output_image_id] selected_output_image = st.session_state["output_images"][output_image_id]
break break
output_image_id += 1 output_image_id += 1
if selected_output_image is not None: if update and selected_output_image is not None:
st.session_state["input_image"] = selected_output_image st.session_state["input_image"] = selected_output_image
return selected_output_image is not None
def apply_stroke_to_image(stroke_image, image): def apply_stroke_to_image(stroke_image, image):
@@ -260,3 +261,5 @@ with column_output:
image = st.session_state.output_images[image_id] image = st.session_state.output_images[image_id]
progress_bar = st.progress(1.0) progress_bar = st.progress(1.0)
show_output_image(image) show_output_image(image)
if "upload_image" in st.session_state and use_output_image_as_input(update=False):
st.markdown("If you want to use an output image as input image, please delete the uploaded image manually.")