Merge pull request #178 from togethercomputer/Nutlope-patch-1

Update image.ts

Author: Nutlope

examples/image.ts

@@ -3,25 +3,20 @@
 //An example to generate an image and save to a file
 
 import Together from 'together-ai';
-import fs from 'fs';
 
 const together = new Together();
 
 async function main() {
-  //Request the image generation
-  const image = await together.images.create({
-    model: 'runwayml/stable-diffusion-v1-5',
-    prompt: 'space robots',
-    n: 1,
+  // Request the image generation
+  const response = await together.images.create({
+    prompt: 'a flying cat',
+    model: 'black-forest-labs/FLUX.1-schnell',
+    steps: 4,
   });
 
-  //Write the image to a file
-  if (image.data && image.data[0] && image.data[0].b64_json) {
-    let image_data = image.data[0].b64_json;
+  console.log(response);
 
-    const buffer = Buffer.from(image_data, 'base64');
-    fs.writeFileSync('image.jpg', buffer, { encoding: 'base64' });
-  }
+  // console.log(response.data[0]?.url);
 }
 
 main();