Sử dụng Transformers.js trên Hugging Face

Transformers.js là một thư viện JavaScript để chạy 🤗 Transformers trực tiếp trong trình duyệt của bạn, không cần máy chủ! Nó được thiết kế để tương đương về chức năng với thư viện Python gốc, nghĩa là bạn có thể chạy các mô hình được huấn luyện trước giống hệt nhau bằng một API rất tương tự.

Khám phá transformers.js trên Hub

Bạn có thể tìm các model transformers.js bằng cách lọc theo thư viện trên trang models.

Khảo sát nhanh

Việc chuyển đổi từ mã hiện có cực kỳ đơn giản! Giống như thư viện Python, chúng tôi hỗ trợ API pipeline. Pipeline nhóm một mô hình được huấn luyện trước với tiền xử lý đầu vào và hậu xử lý đầu ra, giúp đây là cách dễ nhất để chạy mô hình bằng thư viện.

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}]

Bạn cũng có thể sử dụng một model khác bằng cách chỉ định model id hoặc path như đối số thứ hai của hàm pipeline. Ví dụ:

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

Tham khảo tài liệu để xem danh sách đầy đủ các tác vụ và mô hình được hỗ trợ.

Cài đặt

Để cài đặt qua NPM, hãy chạy:

npm i @huggingface/transformers

Để biết thêm thông tin, bao gồm cách sử dụng trong JS thuần (không có bundler) thông qua CDN hoặc lưu trữ tĩnh, hãy tham khảo README.

Tài nguyên bổ sung

• Tệp lưu trữ repository của Transformers.js
• Tài liệu Transformers.js
• Demo Transformers.js