mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 04:11:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

Adding DeepDanbooru to the interrogation API

Browse source
This commit is contained in:
snowmeow2 2022-11-07 02:32:06 +08:00
parent 32c0eab895
commit 67c8e11be7
2 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

1
modules/api/models.py
View file

@ -170,6 +170,7 @@ class ProgressResponse(BaseModel):
class InterrogateRequest(BaseModel):
image: str = Field(default="", title="Image", description="Image to work on, must be a Base64 string containing the image's data.")
model: str = Field(default="clip", title="Model", description="The interrogate model used.")
class InterrogateResponse(BaseModel):
caption: str = Field(default=None, title="Caption", description="The generated caption for the image.")