머신러닝 앱 만들기

머신러닝 앱 만들기

머신러닝 앱을 빠르고 쉽게 만들고 공유할 수 있는 Gradio 라이브러리를 Pipeline에 연결하면, 추론용 간단한 인터페이스를 금방 만들 수 있어요. 추론을 위한 간단한 인터페이스를 만들고 싶다면 이 조합이 딱 좋습니다.

출처: 문서

본문

시작하기 전에 Gradio가 설치되어 있는지 확인해 주세요.

!pip install gradio

먼저 작업(task)에 맞는 파이프라인을 만들고, 이를 Gradio의 Interface.from_pipeline 함수에 넘겨 인터페이스를 만듭니다. Gradio가 Pipeline에 적합한 입력·출력 컴포넌트를 자동으로 결정해 줘요.

launch를 추가하면 웹 서버를 만들고 앱을 시작합니다.

from transformers import pipeline
import gradio as gr

pipeline = pipeline("image-classification", model="google/vit-base-patch16-224")
gr.Interface.from_pipeline(pipeline).launch()

웹 앱은 기본적으로 로컬 서버에서 실행돼요. 다른 사용자와 앱을 공유하려면 launch에서 share=True를 설정해 임시 공개 링크를 생성하면 됩니다. 더 영구적인 방법으로는 Hugging Face Spaces에 앱을 호스팅하는 것을 추천해요.

gr.Interface.from_pipeline(pipeline).launch(share=True)

아래 Space는 위 코드로 만들어진 앱을 Spaces에 호스팅한 것입니다.