diff options
| author | Ryan Voots <simcop2387@simcop2387.info> | 2022-10-17 16:49:33 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-18 05:44:50 +0000 |
| commit | c3851a853d99ad35ccedcdd8dbeb6cfbe273439b (patch) | |
| tree | b8f8a8f42f4184a4114ac696421cabde4332c8b8 /modules/api | |
| parent | f29b16bad19b6332a15b2ef439864d866277fffb (diff) | |
| download | stable-diffusion-webui-gfx803-c3851a853d99ad35ccedcdd8dbeb6cfbe273439b.tar.gz stable-diffusion-webui-gfx803-c3851a853d99ad35ccedcdd8dbeb6cfbe273439b.tar.bz2 stable-diffusion-webui-gfx803-c3851a853d99ad35ccedcdd8dbeb6cfbe273439b.zip | |
Re-use webui fastapi application rather than requiring one or the other, not both.
Diffstat (limited to 'modules/api')
| -rw-r--r-- | modules/api/api.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/modules/api/api.py b/modules/api/api.py index ce72c5ee..8781cd86 100644 --- a/modules/api/api.py +++ b/modules/api/api.py @@ -2,15 +2,13 @@ from modules.api.processing import StableDiffusionProcessingAPI from modules.processing import StableDiffusionProcessingTxt2Img, process_images import modules.shared as shared import uvicorn -from fastapi import FastAPI, Body, APIRouter +from fastapi import Body, APIRouter from fastapi.responses import JSONResponse from pydantic import BaseModel, Field, Json import json import io import base64 -app = FastAPI() - class TextToImageResponse(BaseModel): images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.") parameters: Json @@ -18,7 +16,7 @@ class TextToImageResponse(BaseModel): class Api: - def __init__(self): + def __init__(self, app): self.router = APIRouter() app.add_api_route("/v1/txt2img", self.text2imgapi, methods=["POST"]) |