huggingFace es similar a Github pero para proyectos de AI
Registro
- Registrate
Crea un Token
- ve en la seccion de Tokens en tu perfil: https://huggingface.co/settings/tokens
- cre un token, coloca cualquier nombre y listo obtendrs un token para poder conectarnos
Ejemplo practico
Instalaremos los siguientes paquetes
- HuggingFace Inference
- dotenv
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);
}