huggingFace es similar a Github pero para proyectos de AI

Registro

  1. Registrate

Crea un Token

  1. ve en la seccion de Tokens en tu perfil: https://huggingface.co/settings/tokens
  2. cre un token, coloca cualquier nombre y listo obtendrs un token para poder conectarnos

Ejemplo practico

Instalaremos los siguientes paquetes

npm install @huggingface/inference dotenv

Usar modelos

Si quieres usar cualquier modelo en el Hub de Hugging Face puedes ir a la seccion Modelos y encontrar una enorme cantidad de modulos que puedes llamar desde Nodejs.

En mi caso usare un modelo de "Imagen a Text"

import { HfInference } from "@huggingface/inference";
import dotenv from "dotenv";

dotenv.config();

console.log(process.env.HF_ACCESS_TOKEN);

const HF_ACCESS_TOKEN = process.env.HF_ACCESS_TOKEN;

const inference = new HfInference(HF_ACCESS_TOKEN);

const model = "Salesforce/blip-image-captioning-large";
const imageURL =
  "https://www.amny.com/wp-content/uploads/2020/07/GettyImages-153787739.jpg";

const response = await fetch(imageURL);
const blob = await response.blob();

console.log(blob);
try {
  const result = await inference.imageToText({
    data: blob,
    model,
  });
  console.log(result);
} catch (error) {
  console.error(error);
}

Translate

import { HfInference } from "@huggingface/inference";
import dotenv from "dotenv";

dotenv.config();

const HF_ACCESS_TOKEN = process.env.HF_ACCESS_TOKEN;
const inference = new HfInference(HF_ACCESS_TOKEN);
const model = "facebook/nllb-200-distilled-600M";

try {
  const result = await inference.translation({
    model,
    inputs: "Hello My name is John Doe and I am a software engineer",
    parameters: {
      src_lang: "en",
      tgt_lang: "es",
    },
  });
  console.log(result);
} catch (error) {
  console.error(error);
}

Más Recursos