diff options
| author | snowmeow2 <snowmeow2@example.com> | 2022-11-07 02:32:06 +0800 |
|---|---|---|
| committer | snowmeow2 <snowmeow2@example.com> | 2022-11-07 02:32:06 +0800 |
| commit | 67c8e11be74180be19341aebbd6a246c37a79fbb (patch) | |
| tree | bc0d627cdb67e646d8508cb79928eadc715a1903 /modules/api/models.py | |
| parent | 32c0eab89538ba3900bf499291720f80ae4b43e5 (diff) | |
Adding DeepDanbooru to the interrogation API
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/api/models.py b/modules/api/models.py index 34dbfa16..f9cd929e 100644 --- a/modules/api/models.py +++ b/modules/api/models.py @@ -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.") |