Hugging Face에서 `Transformers.js` 사용하기
Hugging Face에서 Transformers.js 사용하기
Transformers.js는 서버 없이 🤗 Transformers를 브라우저에서 직접 실행하기 위한 JavaScript 라이브러리예요. 기존 Python 라이브러리와 기능적으로 동등하게 설계되어, 거의 같은 API로 같은 사전학습 모델을 실행할 수 있어요.
출처: 문서
본문
Hub에서 transformers.js 탐색하기
모델 페이지에서 라이브러리로 필터링해 transformers.js 모델을 찾을 수 있어요.
빠른 둘러보기
기존 코드에서 변환하는 건 아주 간단해요! Python 라이브러리처럼 pipeline API를 지원해요. 파이프라인은 사전학습 모델을 입력 전처리·출력 후처리와 함께 묶어서, 라이브러리로 모델을 실행하는 가장 쉬운 방법이에요.
Python (원본):
from transformers import pipeline
# Allocate a pipeline for sentiment-analysis
pipe = pipeline('sentiment-analysis')
out = pipe('I love transformers!')
# [{'label': 'POSITIVE', 'score': 0.999806941}]
Javascript (우리 것):
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 함수의 두 번째 인수로 모델 id나 경로를 지정해 다른 모델을 쓸 수도 있어요. 예:
// Use a different model for sentiment-analysis
let pipe = await pipeline('sentiment-analysis', 'nlptown/bert-base-multilingual-uncased-sentiment');
지원되는 작업과 모델의 전체 목록은 문서를 참고하세요.
설치
NPM으로 설치하려면:
npm i @huggingface/transformers
CDN이나 정적 호스팅을 통해 번들러 없이 vanilla JS에서 쓰는 방법을 포함한 더 많은 정보는 README를 참고하세요.