X-Git-Url: https://git.cworth.org/git?a=blobdiff_plain;f=generate-image.py;h=3a2d3f3f65593fdd03a4b308af3ebd412295abf0;hb=b2d238eb845241795fce777cd5a24960a3006e79;hp=229428f3163d284276f653f5ed0c0dd9e5c5618a;hpb=53806afc9bf4bc0cfdd069a6de63db820b004d4a;p=zombocom-ai diff --git a/generate-image.py b/generate-image.py index 229428f..3a2d3f3 100755 --- a/generate-image.py +++ b/generate-image.py @@ -39,7 +39,7 @@ def save_artifact(artifact, prompt): img = Image.open(io.BytesIO(artifact.binary)) img.save(filename) - return {"seed": seed, "prompt": prompt, + return {"code": seed, "prompt": prompt, "filename": filename.removeprefix(DOCUMENT_ROOT)} def main() -> None: