Skip to main content

实践-端对端 MLOps Huggingface spaces

· 2 min read

框架​

使用 Github CI 推送 模型 到 Huggingface Spacers。

准备 HuggingFace Space​

setup Huggingface space and Github space

准备 GitHub Code Space​

enter and setup codespace

virtualenv ~/.venv
vim ~/.bashrc
# add source ~/.venv/bin/activate

保证 pyhton 的环境配置在 terminals 中一致。

代码开发​

代码库: hugging-face-demo

CI/CD 设置​

通过 Git Action 通过 GitHub 代码到 Hugging Face，通过README.md 文件将配置信息传递给 Hugging Face，使用 gradio 作为前端引擎创建画面。

GitHub Action: 需提前设置 GitHub Repository Secret

name: Sync to Hugging Face hub
on:
  push:
    branches: [main]

  # Run this workflow manually from the Action tab
  workflow_dispatch:

jobs:
  sync-to-hub:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v2
        with:
          fetch-depth: 0
      - name: Add remote
        env:
          HF: ${{secrets.HUGGING_FACE_ACCESS_TOKEN}}
        run: git remote add space https://xiaokatech:[email protected]/spaces/xiaokatech/hugging-face-demo
      - name: Push to hub
        env:
          HF: ${{secrets.HUGGING_FACE_ACCESS_TOKEN}}
        run: git push --force https://xiaokatech:[email protected]/spaces/xiaokatech/hugging-face-demo main


做一些相关 debugs.

结果​

总结中文效果不佳：

可以总结短篇英文：

如果文本长一点就会出错，免费的还要什么自行车：