Merge pull request #114 from daily-co/jpt/fal-updates

Updated Fal.ai service to take a params model and allow for model string param
This commit is contained in:
Aleix Conchillo Flaqué
2024-04-11 00:47:33 +08:00
committed by GitHub
10 changed files with 60 additions and 29 deletions

View File

@@ -83,13 +83,12 @@ class TTSService(AIService):
class ImageGenService(AIService):
def __init__(self, image_size, **kwargs):
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.image_size = image_size
# Renders the image. Returns an Image object.
@abstractmethod
async def run_image_gen(self, sentence: str) -> tuple[str, bytes, tuple[int, int]]:
async def run_image_gen(self, prompt: str) -> tuple[str, bytes, tuple[int, int]]:
pass
async def process_frame(self, frame: Frame) -> AsyncGenerator[Frame, None]: