피드로 돌아가기
Using & Mixing Hugging Face Models with Gradio 2.0
Hugging Face BlogHugging Face Blog
Backend

Gradio 2.0이 1줄의 코드로 Hugging Face 모델을 GUI 인터페이스로 로드 및 데모 가능하도록 지원

Using & Mixing Hugging Face Models with Gradio 2.0

2021년 5월 25일5beginner

Context

머신러닝 모델을 프로그래머가 아닌 비프로그래머나 팀 전체와 공유해야 하는 상황이 증가했으나, 기존에는 모델 로드 후 프로그래매틱하게만 사용 가능했습니다. Hugging Face Model Hub에 10,000개 이상의 모델이 있지만 접근성이 제한적이었습니다.

Technical Solution

  • Gradio 2.0에서 Interface 클래스를 활용해 Hugging Face 모델을 1줄의 코드로 GUI 인터페이스 생성: gr.Interface.from_pretrained(model_name)
  • Hugging Face의 호스팅 Inference API를 기본 백엔드로 사용하거나, transformers 라이브러리를 설치해 로컬에서 모델 실행 가능
  • 여러 모델을 병렬로 로드하여 비교 가능: 서로 다른 4개의 텍스트 생성 모델을 동시에 실행하고 결과 비교
  • 여러 모델을 직렬로 연결하여 파이프라인 구성 가능: 예시로 3줄의 코드로 Finnish 뉴스 번역 및 요약 애플리케이션 구현
  • Interface 클래스의 기본 파라미터를 override하여 데모 커스터마이징 지원

Key Takeaway

Gradio 2.0은 머신러닝 모델을 독립적인 소프트웨어가 아닌 조합 가능한 컴포넌트로 취급하여, 복잡한 멀티 모델 애플리케이션을 최소한의 코드로 빠르게 구축할 수 있게 했습니다.


머신러닝 모델을 팀이나 비기술 사용자와 공유해야 하는 상황에서 Gradio의 Interface 클래스를 사용하면 몇 줄의 코드만으로 웹 기반 GUI 데모를 구축 및 배포할 수 있으며, 모델 선택 비교나 데이터 전처리-예측 파이프라인을 손쉽게 구성할 수 있습니다.

원문 읽기