Aller au contenu

Transformer.js

AI generated translation.

HuggingFace Transformers.js est une bibliothèque JavaScript qui vous permet d’exécuter des modèles pré-entraînés localement sur votre machine. La bibliothèque utilise onnxruntime pour exploiter les capacités CPU/GPU de votre matériel.

Dans ce guide, nous allons montrer comment créer des résumés à l’aide de la bibliothèque Transformers.js.

L’extrait ci-dessous importe la bibliothèque Transformers.js et charge le pipeline et le modèle de synthèse. Vous pouvez spécifier un nom de modèle ou laisser la bibliothèque choisir le dernier et le meilleur.

import { pipeline } from "@genaiscript/runtime"
const summarizer = await pipeline("summarization")

L’allocation et le chargement du modèle peuvent prendre un certain temps, il est donc préférable de le faire au début de votre script et une seule fois.

Le pipeline de synthèse prend un seul argument, le contenu à résumer. Il renvoie un tableau de résumés que nous devons déballer pour accéder au texte final du résumé. C’est ce que nous faisons ci-dessous et summary_index contient le texte du résumé.

const [summary] = await summarizer(content)
// @ts-ignore
const { summary_text } = summary