feat: support creating a HuggingFace model (#113)

Because

-  support creating a classification model from HuggingFace repository

This commit

- add HuggingFace model definition and HuggingFace model creation

Dockerfile

@@ -34,6 +34,7 @@ RUN pip3 install --no-cache --upgrade pip setuptools
 RUN apk add --no-cache libffi-dev build-base py3-pip python3-dev
 RUN apk add --no-cache libgit2-dev py3-pygit2
 RUN pip3 install dvc[gs]
+RUN pip3 install transformers pillow torch onnxruntime
 
 WORKDIR /${SERVICE_NAME}
 

Dockerfile.dev

@@ -15,4 +15,13 @@ COPY . .
 
 RUN go install github.com/cosmtrek/air@latest
 
+RUN apt-get update
+RUN apt-get -y install python3
+RUN apt-get -y install python3-setuptools
+RUN apt-get -y install python3-pip
+RUN pip3 install dvc[gs]
+RUN pip3 install transformers pillow torch onnxruntime
+RUN apt-get -y install git git-lfs
+
+
 ENTRYPOINT ["tail", "-f", "/dev/null"]

assets/huggingface-vit-template/README.md

@@ -0,0 +1,7 @@
+---
+Task: Classification
+Tags:
+  - Classification
+---
+
+# Template repo for HuggingFace model

assets/huggingface-vit-template/huggingface-infer/config.pbtxt

@@ -0,0 +1,4 @@
+name: "huggingface-infer"
+platform: "onnxruntime_onnx"
+max_batch_size: 32
+version_policy: { all { }}