Image Generation
ARK Platform Example / Image Generation
-
# NOTE: This code will output a "knight_in_space.png" file with the generated image. import openai import base64 from PIL import Image from io import BytesIO ark_base_url = "https://api.ark-labs.cloud/api/v1" ark_api_key = "API_KEY" client = openai.OpenAI(api_key=ark_api_key, base_url=ark_base_url) print("Generating image...") response = client.images.generate( model="gpt-image-1", prompt="A brave knight in a futuristic space suit floating near a small rocket, lost in deep space, photorealistic CGI.", n=1, size="1024x1024", response_format="b64_json", extra_body={"steps": 10} ) image_b64 = response.data[0].b64_json image_bytes = base64.b64decode(image_b64) image = Image.open(BytesIO(image_bytes)) image.save("knight_in_space.png") print("Image saved as knight_in_space.png")