diff --git a/modules/api/api.py b/modules/api/api.py index a96700283..ee32648d3 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -77,8 +77,8 @@ class Api: self.add_api_route("/sdapi/v1/process-batch", self.process.extras_batch_images_api, methods=["POST"], response_model=models.ResProcessBatch, tags=["Processing"]) self.add_api_route("/sdapi/v1/extra-batch-images", self.process.extras_batch_images_api, methods=["POST"], response_model=models.ResProcessBatch, tags=["Processing"]) self.add_api_route("/sdapi/v1/preprocess", self.process.post_preprocess, methods=["POST"], response_model=models.ResPreprocess, tags=["Processing"]) - self.add_api_route("/sdapi/v1/mask", self.process.post_mask, methods=["POST"], response_model=models.ResMask, tags=["Processing"]) - self.add_api_route("/sdapi/v1/detect", self.process.post_detect, methods=["POST"], response_model=models.ResFace, tags=["Processing"]) + self.add_api_route("/sdapi/v1/mask", self.process.post_mask, methods=["POST"], response_model=process.ResMask, tags=["Processing"]) + self.add_api_route("/sdapi/v1/detect", self.process.post_detect, methods=["POST"], response_model=process.ResFace, tags=["Processing"]) self.add_api_route("/sdapi/v1/detail", self.process.post_detail, methods=["POST"], response_model=models.ResDetail, tags=["Processing"]) self.add_api_route("/sdapi/v1/prompt-enhance", self.process.post_prompt_enhance, methods=["POST"], response_model=models.ResPromptEnhance, tags=["Generation"]) diff --git a/modules/api/gallery.py b/modules/api/gallery.py index 39c6b850a..d3b2d7d32 100644 --- a/modules/api/gallery.py +++ b/modules/api/gallery.py @@ -210,7 +210,7 @@ def register_api(api): # register api debug(f'Browser folders allow: {path}') shared.demo.allowed_paths.append(path) debug(f'Browser folders: {unique_folders}') - return JSONResponse(content=unique_folders) + return unique_folders # @app.get("/sdapi/v1/browser/thumb", response_model=dict) async def get_thumb(file: str, exif: bool = False):