aboutsummaryrefslogtreecommitdiff
path: root/modules/api
diff options
context:
space:
mode:
authorAUTOMATIC1111 <16777216c@gmail.com>2023-04-29 09:30:33 +0300
committerGitHub <noreply@github.com>2023-04-29 09:30:33 +0300
commita33d49cc572214494e1e5f1fdcc4f422b370eae1 (patch)
treedd4e0baa7eb4a256f2bd74dd917abd56e604a37a /modules/api
parentc19618f37059b425b1e53429ad8def2caa78cdec (diff)
parent7fc10e04456878ac4e0881dbfe56c8966e22f40d (diff)
Merge branch 'dev' into fix_batch_processing
Diffstat (limited to 'modules/api')
-rw-r--r--modules/api/api.py12
1 files changed, 3 insertions, 9 deletions
diff --git a/modules/api/api.py b/modules/api/api.py
index 518b2a61..5ed670e9 100644
--- a/modules/api/api.py
+++ b/modules/api/api.py
@@ -6,7 +6,6 @@ import uvicorn
import gradio as gr
from threading import Lock
from io import BytesIO
-from gradio.processing_utils import decode_base64_to_file
from fastapi import APIRouter, Depends, FastAPI, Request, Response
from fastapi.security import HTTPBasic, HTTPBasicCredentials
from fastapi.exceptions import HTTPException
@@ -395,16 +394,11 @@ class Api:
def extras_batch_images_api(self, req: ExtrasBatchImagesRequest):
reqDict = setUpscalers(req)
- def prepareFiles(file):
- file = decode_base64_to_file(file.data, file_path=file.name)
- file.orig_name = file.name
- return file
-
- reqDict['image_folder'] = list(map(prepareFiles, reqDict['imageList']))
- reqDict.pop('imageList')
+ image_list = reqDict.pop('imageList', [])
+ image_folder = [decode_base64_to_image(x.data) for x in image_list]
with self.queue_lock:
- result = postprocessing.run_extras(extras_mode=1, image="", input_dir="", output_dir="", save_output=False, **reqDict)
+ result = postprocessing.run_extras(extras_mode=1, image_folder=image_folder, image="", input_dir="", output_dir="", save_output=False, **reqDict)
return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1])