Unverified Commit 38f1c818 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #9513 from ilya-khadykin/fix_batch_processing

fix(extras): fix batch image processing on 'Extras\Batch Process' tab
parents 7fc10e04 a33d49cc
Loading
Loading
Loading
Loading
+4 −2
Original line number Diff line number Diff line
import os
import tempfile
from typing import List

from PIL import Image

@@ -6,7 +8,7 @@ from modules import shared, images, devices, scripts, scripts_postprocessing, ui
from modules.shared import opts


def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir, show_extras_results, *args, save_output: bool = True):
def run_postprocessing(extras_mode, image, image_folder: List[tempfile.NamedTemporaryFile], input_dir, output_dir, show_extras_results, *args, save_output: bool = True):
    devices.torch_gc()

    shared.state.begin()
@@ -22,7 +24,7 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir,
                image = img
                fn = ''
            else:
                image = Image.open(img)
                image = Image.open(os.path.abspath(img.name))
                fn = os.path.splitext(img.orig_name)[0]

            image_data.append(image)
+1 −1
Original line number Diff line number Diff line
@@ -13,7 +13,7 @@ def create_ui():
                    extras_image = gr.Image(label="Source", source="upload", interactive=True, type="pil", elem_id="extras_image")

                with gr.TabItem('Batch Process', elem_id="extras_batch_process_tab") as tab_batch:
                    image_batch = gr.File(label="Batch Process", file_count="multiple", interactive=True, type="file", elem_id="extras_image_batch")
                    image_batch = gr.Files(label="Batch Process", interactive=True, elem_id="extras_image_batch")

                with gr.TabItem('Batch from Directory', elem_id="extras_batch_directory_tab") as tab_batch_dir:
                    extras_batch_input_dir = gr.Textbox(label="Input directory", **shared.hide_dirs, placeholder="A directory on the same machine where the server is running.", elem_id="extras_batch_input_dir")