mirror of
https://github.com/openvinotoolkit/stable-diffusion-webui.git
synced 2024-12-14 22:53:25 +03:00
Remove folder endpoint
This commit is contained in:
parent
8320963dcb
commit
b2e0d8ba78
@ -32,7 +32,6 @@ class Api:
|
||||
self.app.add_api_route("/sdapi/v1/img2img", self.img2imgapi, methods=["POST"], response_model=ImageToImageResponse)
|
||||
self.app.add_api_route("/sdapi/v1/extra-single-image", self.extras_single_image_api, methods=["POST"], response_model=ExtrasSingleImageResponse)
|
||||
self.app.add_api_route("/sdapi/v1/extra-batch-images", self.extras_batch_images_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)
|
||||
self.app.add_api_route("/sdapi/v1/extra-folder-images", self.extras_folder_processing_api, methods=["POST"], response_model=ExtrasBatchImagesResponse)
|
||||
|
||||
def text2imgapi(self, txt2imgreq: StableDiffusionTxt2ImgProcessingAPI):
|
||||
sampler_index = sampler_to_index(txt2imgreq.sampler_index)
|
||||
@ -126,14 +125,6 @@ class Api:
|
||||
|
||||
return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1])
|
||||
|
||||
def extras_folder_processing_api(self, req:ExtrasFoldersRequest):
|
||||
reqDict = setUpscalers(req)
|
||||
|
||||
with self.queue_lock:
|
||||
result = run_extras(extras_mode=2, image=None, image_folder=None, **reqDict)
|
||||
|
||||
return ExtrasBatchImagesResponse(images=list(map(encode_pil_to_base64, result[0])), html_info=result[1])
|
||||
|
||||
def pnginfoapi(self):
|
||||
raise NotImplementedError
|
||||
|
||||
|
@ -149,7 +149,3 @@ class ExtrasBatchImagesRequest(ExtrasBaseRequest):
|
||||
|
||||
class ExtrasBatchImagesResponse(ExtraBaseResponse):
|
||||
images: list[str] = Field(title="Images", description="The generated images in base64 format.")
|
||||
|
||||
class ExtrasFoldersRequest(ExtrasBaseRequest):
|
||||
input_dir: str = Field(title="Input directory", description="Directory path from where to take the images")
|
||||
output_dir: str = Field(title="Output directory", description="Directory path to put the processsed images into")
|
||||
|
Loading…
Reference in New Issue
Block a user