hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 19:22:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

gradio and FastAPI

Browse Source
This commit is contained in:
arcticfaded
2022-10-18 06:51:53 +00:00
parent 1df3ff25e6
commit 8d5d863a9d
2 changed files with 16 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modules/api/api.py
View File

@@ -16,9 +16,11 @@ class TextToImageResponse(BaseModel):
class Api:
def __init__(self, app):
def __init__(self, app, queue_lock):
self.router = APIRouter()
app.add_api_route("/sdapi/v1/txt2img", self.text2imgapi, methods=["POST"])
self.app = app
self.queue_lock = queue_lock
self.app.add_api_route("/sdapi/v1/txt2img", self.text2imgapi, methods=["POST"])
def text2imgapi(self, txt2imgreq: StableDiffusionProcessingAPI ):
populate = txt2imgreq.copy(update={ # Override __init__ params
@@ -30,7 +32,8 @@ class Api:
)
p = StableDiffusionProcessingTxt2Img(**vars(populate))
# Override object param
processed = process_images(p)
with self.queue_lock:
processed = process_images(p)
b64images = []
for i in processed.images:
@@ -52,5 +55,5 @@ class Api:
raise NotImplementedError
def launch(self, server_name, port):
app.include_router(self.router)
uvicorn.run(app, host=server_name, port=port)
self.app.include_router(self.router)
uvicorn.run(self.app, host=server_name, port=port)