Socket
Book a DemoInstallSign in
Socket

node-replicate

Package Overview
Dependencies
Maintainers
1
Versions
15
Alerts
File Explorer

Advanced tools

Socket logo

Install Socket

Detect and block malicious and high-risk dependencies

Install

node-replicate

A NodeJS client for Replicate.

Source
npmnpm
Version
1.0.9
Version published
Weekly downloads
68
-9.33%
Maintainers
1
Weekly downloads
 
Created
Source

node-replicate

A NodeJS client for Replicate.

import replicate from "node-replicate"

const prediction = await replicate
  .model(
    "stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf",
   )
  .predict({
    prompt: "an astronaut riding on a horse",
  })

console.log(prediction.output)

// [ "https://replicate.delivery/pbxt/f4nlztv3uz1iFC4AEf2wBYQGTezdVeysvtZUtwfsvZOJDN6AC/out-0.png" ]

Introduction

Replicate is an online platform for running machine learning models in the cloud. This package implements a lightweight client for their anonymous API, allowing you to run Stable Diffusion, Midjourney and other cutting-edge models with just a few lines of code 😊👌.

Features

  • Run Replicate models anonymously 👻.
  • Track pending predictions ⌛.
  • Very lightweight - under 100 lines of code ⚡.

Installation

Install with npm.

npm i node-replicate

Usage

To run a Replicate model, pass its identifier to replicate.model() and then invoke predict() asynchronously.

import replicate from "node-replicate"

const prediction = await replicate
  .model(
    "stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf",
   )
  .predict({
    prompt: "an astronaut riding on a horse",
  })

console.log(prediction.output)

// [ "https://replicate.delivery/pbxt/f4nlztv3uz1iFC4AEf2wBYQGTezdVeysvtZUtwfsvZOJDN6AC/out-0.png" ]

You can also track a pending predictions by passing an onUpdate() callback.

import replicate from "node-replicate"

const prediction = await replicate
  .model(
    "stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf",
   )
  .predict(
    {
      prompt: "an astronaut riding on a horse",
    },
    {
      onUpdate(prediction) {
        console.log(prediction.status)
      }
    },
  )

console.log(prediction.output)

// [ "https://replicate.delivery/pbxt/f4nlztv3uz1iFC4AEf2wBYQGTezdVeysvtZUtwfsvZOJDN6AC/out-0.png" ]

Examples

1. Visual Question-answering with blip

import replicate from "node-replicate"

const prediction = await replicate
  .model(
    "salesforce/blip:2e1dddc8621f72155f24cf2e0adbde548458d3cab9f00c0139eea840d0ac4746",
   )
  .predict({
    image: 'https://replicate.delivery/pbxt/IVSaMZb8iBkELQvQya84wz5i1YfQC1HxrtSfSaL4QRTtsOlP/cat.jpg',
    question: "What color is the cat?",
    task: "visual_question_answering",
  })

console.log(prediction.output)

// "Answer: orange"

2. Image Style Transfer with clipstyler

import replicate from "node-replicate"

const prediction = await replicate
  .model(
    "salesforce/blip:2e1dddc8621f72155f24cf2e0adbde548458d3cab9f00c0139eea840d0ac4746",
   )
  .predict({
    image: "https://replicate.delivery/pbxt/IVSrp3308R8Uq0sJx6yAozDSuLswkkq6IlaOS5liUI7TCwAU/cat.jpg",
    iterations: 100,
    text: "made from leaves",
  })

console.log(prediction.output)

// [ ... ]

Contributing

Have a feature you'd like to see added? Create a pull request or open an issue.

Keywords

ai

FAQs

Package last updated on 20 Mar 2023

Did you know?

Socket

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts