diff options
| author | Bruno Seoane <brunoseoaneamarillo@gmail.com> | 2022-10-22 20:24:04 -0300 |
|---|---|---|
| committer | Bruno Seoane <brunoseoaneamarillo@gmail.com> | 2022-10-22 20:24:04 -0300 |
| commit | b02926df1393df311db734af149fb9faf4389cbe (patch) | |
| tree | 9288a5af081fd8e611270cf2db39bce0532acb63 /modules/api/models.py | |
| parent | 1b4d04737ac513cbd55958bb60a4f85166f3484b (diff) | |
Moved moodels to their own file and extracted base64 conversion to its own function
Diffstat (limited to 'modules/api/models.py')
| -rw-r--r-- | modules/api/models.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/modules/api/models.py b/modules/api/models.py new file mode 100644 index 00000000..a7d247d8 --- /dev/null +++ b/modules/api/models.py @@ -0,0 +1,8 @@ +from pydantic import BaseModel, Field, Json + +class TextToImageResponse(BaseModel): + images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.") + parameters: Json + info: Json + +
\ No newline at end of file |