Adding DeepDanbooru to the interrogation API

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

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.")