| import base64, json, urllib.request | |
| payload = { | |
| "model": "/model", | |
| "prompt": "A realistic photograph of a red ceramic teapot on a wooden table beside a window, soft morning light, detailed wood grain.", | |
| "negative_prompt": " ", "size": "512x512", | |
| "num_inference_steps": 50, "guidance_scale": 4.0, "seed": 42, | |
| "response_format": "b64_json", "format": "png", | |
| } | |
| request = urllib.request.Request( | |
| "http://127.0.0.1:8000/v1/images/generations", | |
| data=json.dumps(payload).encode(), headers={"Content-Type": "application/json"}) | |
| with urllib.request.urlopen(request, timeout=600) as response: | |
| result = json.load(response) | |
| with open("sample.png", "wb") as output: | |
| output.write(base64.b64decode(result["data"][0]["b64_json"])) | |
| print("Saved sample.png") | |