midjourney/src/verify.human.ts

Node.js client for the unofficial MidJourney API.

import { HfInference } from "@huggingface/inference";
import { MJConfig } from "./interfaces";
export class VerifyHuman {
  private inference: HfInference;

  constructor(public config: MJConfig) {
    const { HuggingFaceToken } = config;
    if (HuggingFaceToken === "" || HuggingFaceToken) {
      throw new Error("HuggingFaceToken is required");
    }
    this.inference = new HfInference(HuggingFaceToken);
  }

  async verify(imageUri: string, categories: string[]) {
    console.log("verify----start", imageUri, categories);
    const imageCates = await this.inference.imageClassification({
      data: await (await this.config.fetch(imageUri)).blob(),
      model: "google/vit-base-patch16-224",
    });
    console.log("verify----response", { imageCates });
    for (const imageCate of imageCates) {
      const { label } = imageCate;
      for (const category of categories) {
        if (label.includes(category)) {
          return category;
        }
      }
    }
  }
}