Использование Transformers.js на Hugging Face

Transformers.js — это библиотека JavaScript для запуска 🤗 Transformers непосредственно в браузере, без необходимости использовать сервер! Она спроектирована так, чтобы быть функционально эквивалентной исходной библиотеке на Python, что позволяет запускать те же предобученные модели с помощью очень похожего API.

Исследование transformers.js в Hub

Модели transformers.js можно найти на странице моделей, отфильтровав по библиотеке.

Краткий обзор

Перевод с существующего кода выполняется очень просто! Как и в библиотеке Python, мы поддерживаем API pipeline. Пайплайны объединяют предобученную модель с предобработкой входных данных и постобработкой выходных, что делает их самым простым способом запуска моделей с помощью библиотеки.

from transformers import pipeline

# Allocate a pipeline for sentiment-analysis
pipe = pipeline('sentiment-analysis')

out = pipe('I love transformers!')
# [{'label': 'POSITIVE', 'score': 0.999806941}]

import { pipeline } from '@huggingface/transformers';

// Allocate a pipeline for sentiment-analysis
let pipe = await pipeline('sentiment-analysis');

let out = await pipe('I love transformers!');
// [{'label': 'POSITIVE', 'score': 0.999817686}]

Вы также можете использовать другую модель, указав идентификатор или путь к модели в качестве второго аргумента функции pipeline. Например:

// Use a different model for sentiment-analysis
let pipe = await pipeline('sentiment-analysis', 'nlptown/bert-base-multilingual-uncased-sentiment');

Список поддерживаемых задач и моделей см. в документации.

Установка

Чтобы установить через NPM, выполните:

npm i @huggingface/transformers

Для получения дополнительной информации, в том числе о том, как использовать его в чистом JS (без сборщика) через CDN или статический хостинг, обратитесь к README.

Дополнительные ресурсы

• Репозиторий Transformers.js
• Документация Transformers.js
• Демо Transformers.js