Stable DiffusionのAPIをNode.jsから使ってネコ画像生成を試してみる

最近はChatGPT周りが話題でOpenAI側のAPIを触ることが多かったですがStable DiffusionのAPIもDreamStudio経由で使える模様だったので触ってみました。

APIキーの取得

クラメソさんの記事を参考にDreamStudioからAPIキーを取得します。

JavaScript向けSDKを触ってみる

こちらを使ってみます。

npm i stability-client

サンプルコードを元にこちらで、prompt部分にテキストで命令します。
富裕層が飼ってるネコ としました。

Cats owned by wealthy people

const generateAsync = require('stability-client').generateAsync

const main = async () => {
    try {
    const { res, images } = await generateAsync({
        prompt: 'Cats owned by wealthy people', //富裕層が飼ってるネコ
        apiKey: 'APIキー',
    })
    console.log(images)
    } catch (e) {
        console.log(e)
    }
}
main();

画像生成結果

実行すると.outフォルダ内に画像が生成されます。

富裕層が飼ってそうなネコですね。

おまけ

ちなみにOtaku working on his laptop at a maid cafe in Akihabara

秋葉原のメイドカフェでノートPCで作業してるオタク

としたらそれっぽいものが出てきました。