Moved moodels to their own file and extracted base64 conversion to its own function

2022-10-22 20:24:04 -03:00
parent 1b4d04737a
commit b02926df13
2 changed files with 14 additions and 11 deletions
@@ -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
+
+