Update DeepInfraImage.py

2024-04-11 03:52:37 +02:00 · 2024-04-11 03:52:37 +02:00 · deb3898e7a
parent 4271fb9870
commit deb3898e7a
1 changed files with 2 additions and 0 deletions
--- a/g4f/Provider/DeepInfraImage.py
+++ b/g4f/Provider/DeepInfraImage.py
@ -70,5 +70,7 @@ class DeepInfraImage(AsyncGeneratorProvider, ProviderModelMixin):
                await raise_for_status(response)
                data = await response.json()
                images = data["output"] if "output" in data else data["images"]
+                if not images:
+                    raise RuntimeError(f"Response: {data}")
                images = images[0] if len(images) == 1 else images
                return ImageResponse(images, prompt)