zsxkib/wd-image-tagger:2c081a1d – Run with an API on Replicate

Version

You're looking at a specific version of this model. Jump to the model overview.

zsxkib /wd-image-tagger:2c081a1d

Playground API

Input

image

*file

Path to the input image file to be analyzed by the WaifuDiffusion tagger model

model_repo

string

Name of the pre-trained model repository to use for image analysis

Default: "models/wd-swinv2-tagger-v3"

general_thresh

number

(minimum: 0, maximum: 1)

Probability threshold for including general tags in the output (between 0 and 1)

Default: 0.35

general_mcut_enabled

boolean

Whether to use the MCut algorithm to automatically determine the general tags threshold

Default: false

character_thresh

number

(minimum: 0, maximum: 1)

Probability threshold for including character tags in the output (between 0 and 1)

Default: 0.85

character_mcut_enabled

boolean

Whether to use the MCut algorithm to automatically determine the character tags threshold

Default: false

category

string

Category of tags to return in the output

Default: "all_tags"

Run this model in Node.js with one line of code:

npx create-replicate --model=zsxkib/wd-image-tagger

or set up a project from scratch

Install Replicate’s Node.js client library:

npm install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import and set up the client:

import Replicate from "replicate";

const replicate = new Replicate({
  auth: process.env.REPLICATE_API_TOKEN,
});

Run zsxkib/wd-image-tagger using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

const output = await replicate.run(
  "zsxkib/wd-image-tagger:2c081a1ded296dda31ce8f85c50588a4a44e9dc8ad55a350bae6e5596e2d5b63",
  {
    input: {
      category: "all_tags",
      model_repo: "models/wd-swinv2-tagger-v3",
      general_thresh: 0.35,
      character_thresh: 0.85,
      general_mcut_enabled: false,
      character_mcut_enabled: false
    }
  }
);

console.log(output);

To learn more, take a look at the guide on getting started with Node.js.

Install Replicate’s Python client library:

pip install replicate

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Import the client:

import replicate

Run zsxkib/wd-image-tagger using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

output = replicate.run(
    "zsxkib/wd-image-tagger:2c081a1ded296dda31ce8f85c50588a4a44e9dc8ad55a350bae6e5596e2d5b63",
    input={
        "category": "all_tags",
        "model_repo": "models/wd-swinv2-tagger-v3",
        "general_thresh": 0.35,
        "character_thresh": 0.85,
        "general_mcut_enabled": False,
        "character_mcut_enabled": False
    }
)
print(output)

To learn more, take a look at the guide on getting started with Python.

Set the REPLICATE_API_TOKEN environment variable:

export REPLICATE_API_TOKEN=<paste-your-token-here>

Find your API token in your account settings.

Run zsxkib/wd-image-tagger using Replicate’s API. Check out the model's schema for an overview of inputs and outputs.

curl -s -X POST \
  -H "Authorization: Bearer $REPLICATE_API_TOKEN" \
  -H "Content-Type: application/json" \
  -H "Prefer: wait" \
  -d $'{
    "version": "zsxkib/wd-image-tagger:2c081a1ded296dda31ce8f85c50588a4a44e9dc8ad55a350bae6e5596e2d5b63",
    "input": {
      "category": "all_tags",
      "model_repo": "models/wd-swinv2-tagger-v3",
      "general_thresh": 0.35,
      "character_thresh": 0.85,
      "general_mcut_enabled": false,
      "character_mcut_enabled": false
    }
  }' \
  https://api.replicate.com/v1/predictions

To learn more, take a look at Replicate’s HTTP API reference docs.

Output

No output yet! Press "Submit" to start a prediction.