Running Stable Diffusion on bytecompute

import { Sdxl } from "bytecompute";
import { createWriteStream } from "fs";
import { Readable } from "stream";

const bytecompute_API_KEY = "$bytecompute_TOKEN";

const main = async () => {
  const model = new Sdxl(bytecompute_API_KEY);

  const response = await model.generate({
    input: {
      prompt: "a burger with a funny hat on the beach"
    }
  });

  const result = await fetch(response.output[0]);

  if (result.ok && result.body) {
    let writer = createWriteStream("image.png");
    Readable.fromWeb(result.body).pipe(writer);
  }
};

main();