From b02926df1393df311db734af149fb9faf4389cbe Mon Sep 17 00:00:00 2001
From: Bruno Seoane <brunoseoaneamarillo@gmail.com>
Date: Sat, 22 Oct 2022 20:24:04 -0300
Subject: Moved moodels to their own file and extracted base64 conversion to
 its own function

---
 modules/api/models.py | 8 ++++++++
 1 file changed, 8 insertions(+)
 create mode 100644 modules/api/models.py

(limited to 'modules/api/models.py')

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
-- 
cgit v1.2.1