@@ -127,10 +127,14 @@ def replicate_run(callback_url: str, payload: dict[str, Any]):
         if isinstance(image := payload.get('image'), FieldFile):
             payload['image'], temporary_file = ImageInputService.prepare_for_replicate(image)
 
-        return replicate_client.run(
+        output = replicate_client.run(
             ref=callback_url,
             input=payload,
         )
+        if not output:
+            logger.error('Replicate returned empty output for %s: %r', callback_url, output)
+            raise GenerationException
+        return output
     except ModelError as exc:
         prediction_error = getattr(getattr(exc, 'prediction', None), 'error', '') or ''
         error_text = str(exc)