allow webp uploads to img2img tab #991

Sygil-Dev · Sep 16, 2022 · b856a91 · b856a91
1 parent c465891
commit b856a91
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/scripts/img2img.py b/scripts/img2img.py
@@ -483,7 +483,7 @@ def layout():
  image_holder = st.empty()
 
  uploaded_images = st.file_uploader(
- "Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg"],
+ "Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg", "webp"],
  help="Upload an image which will be used for the image to image generation.",
  )
  if uploaded_images:
@@ -494,7 +494,7 @@ def layout():
  mask_holder = st.empty()
 
  uploaded_masks = st.file_uploader(
- "Upload Mask", accept_multiple_files=False, type=["png", "jpg", "jpeg"],
+ "Upload Mask", accept_multiple_files=False, type=["png", "jpg", "jpeg", "webp"],
  help="Upload an mask image which will be used for masking the image to image generation.",
  )
  if uploaded_masks:

diff --git a/scripts/imglab.py b/scripts/imglab.py
@@ -149,7 +149,7 @@ def layout():
 
  col1, col2 = st.columns(2)
  with col1:
- st.session_state['uploaded_file'] = st.file_uploader("Choose an image or images", type=["png", "jpg", "jpeg"],accept_multiple_files=True,on_change=changeImage)
+ st.session_state['uploaded_file'] = st.file_uploader("Choose an image or images", type=["png", "jpg", "jpeg", "webp"],accept_multiple_files=True,on_change=changeImage)
  if 'previewImg' not in st.session_state:
  st.session_state['previewImg'] = st.empty()
  else:

diff --git a/scripts/txt2vid.py b/scripts/txt2vid.py
@@ -548,7 +548,7 @@ def layout():
  height = st.slider("Height:", min_value=64, max_value=2048, value=st.session_state['defaults'].txt2vid.height, step=64)
  cfg_scale = st.slider("CFG (Classifier Free Guidance Scale):", min_value=1.0, max_value=30.0, value=st.session_state['defaults'].txt2vid.cfg_scale, step=0.5, help="How strongly the image should follow the prompt.")
 
- #uploaded_images = st.file_uploader("Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg"],
+ #uploaded_images = st.file_uploader("Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg", "webp"],
  #help="Upload an image which will be used for the image to image generation.") 
  seed = st.text_input("Seed:", value=st.session_state['defaults'].txt2vid.seed, help=" The seed to use, if left blank a random seed will be generated.")
  #batch_count = st.slider("Batch count.", min_value=1, max_value=100, value=st.session_state['defaults'].txt2vid.batch_count, step=1, help="How many iterations or batches of images to generate in total.")

diff --git a/scripts/webui_streamlit_old.py b/scripts/webui_streamlit_old.py
@@ -2432,7 +2432,7 @@ def layout():
  image_holder = st.empty()
 
  uploaded_images = st.file_uploader(
- "Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg"],
+ "Upload Image", accept_multiple_files=False, type=["png", "jpg", "jpeg", "webp"],
  help="Upload an image which will be used for the image to image generation.",
  )
  if uploaded_images:
@@ -2443,7 +2443,7 @@ def layout():
  mask_holder = st.empty()
 
  uploaded_masks = st.file_uploader(
- "Upload Mask", accept_multiple_files=False, type=["png", "jpg", "jpeg"],
+ "Upload Mask", accept_multiple_files=False, type=["png", "jpg", "jpeg", "webp"],
  help="Upload an mask image which will be used for masking the image to image generation.",
  )
  if uploaded_masks: