A little cleanup

2024-01-05 20:41:52 -05:00
parent b48a377b17
commit 7f51c0c9b2
4 changed files with 25 additions and 22 deletions
--- a/src/dailyai/services/ai_services.py
+++ b/src/dailyai/services/ai_services.py
@@ -44,7 +44,7 @@ class TTSService(AIService):
 class ImageGenService(AIService):
    # Renders the image. Returns an Image object.
    @abstractmethod
-    async def run_image_gen(self, sentence) -> tuple[str, bytes]:
+    async def run_image_gen(self, sentence, size) -> tuple[str, bytes]:
        pass