背景信息

當前，開放的模型社區（例如 huggingface、modelscope ）提供了大量的機器學習模型和代碼實現，其對應庫的接口將模型、框架以及相關的處理邏輯進行了封裝，您僅需使用幾行代碼即可直接端到端進行模型訓練和調用等操作，無需考慮復雜的環境依賴、預處理后處理邏輯、框架類型等一系列使用和部署落地模型時的問題。這種生態也是對以Tensorflow、Pytorch為代表的框架-模型形態的升級。

EAS也對該形態進行了優化支持，支持您一鍵部署社區模型。

部署HuggingFace模型

PAI支持將HuggingFace社區官方庫中的tasks對應的模型快速部署為EAS模型在線服務，具體操作步驟如下：

1. 在官方庫tasks中選擇要部署的模型，本文以文本分類模型為例，進入distilbert-base-uncased-finetuned-sst-2-english模型頁面，分別獲取下圖中的MODEL_ID（模型ID）、TASK（模型對應的TASK）、REVISION（模型版本）的值，并保存到本地。

   您可以根據下表中的對應關系找到部署EAS服務時所需的TASK：

   HuggingFace頁面展示的TASK	部署EAS服務時填寫的TASK
   Audio Classification	audio-classification
   Automatic Speech Recognition(ASR)	automatic-speech-recognition
   Feature Extraction	feature-extraction
   Fill Mask	fill-mask
   Image Classification	image-classification
   Question Answering	question-answering
   Summarization	summarization
   Text Classification	text-classification
   Sentiment Analysis	sentiment-analysis
   Text Generation	text-generation
   Translation	translation
   Translation (xx-to-yy)	translation_xx_to_yy
   Text-to-Text Generation	text2text-generation
   Zero-Shot Classification	zero-shot-classification
   Document Question Answering	document-question-answering
   Visual Question Answering	visual-question-answering
   Image-to-Text	image-to-text

2. 在EAS模型在線服務頁面，部署HuggingFace模型。

   1. 進入新建服務頁面，配置以下關鍵參數，其他參數配置詳情，請參見服務部署：控制臺。

      

      參數	描述
      服務名稱	參照界面提示自定義配置服務名稱。
      部署方式	選擇鏡像部署AI-Web應用。
      鏡像選擇	在PAI平臺鏡像列表中選擇huggingface-inference；并根據實際場景選擇對應的鏡像版本。
      環境變量	配置以下參數為步驟1中查詢的結果： MODEL_ID：distilbert-base-uncased-finetuned-sst-2-english。 TASK：text-classification。 REVISION：main。
      運行命令	配置鏡像版本后，系統會自動配置運行命令，您無需修改配置。
      額外系統盤	單擊資源部署信息區域的系統盤配置，將額外系統盤配置為100 GB。

   2. 單擊部署。當服務狀態變為運行中時，表明服務已部署成功。

3. 調用已部署成功的模型服務。

   通過控制臺調用服務

   • 在模型在線服務（EAS）頁面，單擊服務方式列下的查看Web應用，在WebUI頁面驗證模型推理效果。

   • 單擊目標服務操作列下的在線調試，在Body頁簽輸入請求數據，例如：{"data": ["hello"]}，單擊發送請求。

     說明

     文本分類模型輸入的數據格式（{"data": ["XXX"]}）是Gradio框架的/api/predict定義的，如果您使用其他類型的模型，例如圖片分類或語音數據處理等，可以參考/api/predict的定義來構造請求數據。

   通過API調用服務

   1. 單擊服務名稱進入服務詳情頁面，在該頁面單擊查看調用信息。

   2. 在調用信息對話框公網地址調用頁簽，查看訪問地址和Token，并保存到本地。

   3. 通過如下代碼使用API調用服務。

      import requests
      
      resp = requests.post(url="<service_url>",
                    headers={"Authorization": "<token>"},
                    json={"data": ["hello"]})
      
      print(resp.content)
      
      # resp: {"data":[{"label":"POSITIVE","confidences":[{"label":"POSITIVE","confidence":0.9995185136795044}]}],"is_generating":false,"duration":0.280987024307251,"average_duration":0.280987024307251}

      其中：<service_url>和<token>均需要替換為步驟b中獲取的訪問地址和Token。