Add a Dockerfile

.dockerignore

@@ -0,0 +1,6 @@
+.idea/
+.venv/
+.ruff_cache/
+__pycache__/
+
+models/

Dockerfile

@@ -0,0 +1,59 @@
+FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+RUN apt-get update && \
+    apt-get upgrade -y && \
+    apt-get install -y --no-install-recommends \
+    git \
+    git-lfs \
+    wget \
+    curl \
+    # python build dependencies \
+    build-essential \
+    libssl-dev \
+    zlib1g-dev \
+    libbz2-dev \
+    libreadline-dev \
+    libsqlite3-dev \
+    libncursesw5-dev \
+    xz-utils \
+    tk-dev \
+    libxml2-dev \
+    libxmlsec1-dev \
+    libffi-dev \
+    liblzma-dev \
+    # gradio dependencies \
+    ffmpeg \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:${PATH}
+WORKDIR ${HOME}/app
+
+RUN curl https://pyenv.run | bash
+ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+ARG PYTHON_VERSION=3.10.12
+RUN pyenv install ${PYTHON_VERSION} && \
+    pyenv global ${PYTHON_VERSION} && \
+    pyenv rehash && \
+    pip install --no-cache-dir -U pip setuptools wheel && \
+    pip install packaging ninja
+
+COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
+
+COPY --chown=1000 . ${HOME}/app
+ENV PYTHONPATH=${HOME}/app \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
+    SYSTEM=spaces
+
+CMD ["python", "app.py"]

README.md

@@ -24,3 +24,15 @@ uv pip install -r requirements.txt
 ```shell
 python app.py
 ```
+
+## Build Docker image
+
+```shell
+docker build -t radtts-vocos-api .
+```
+
+## Run a container
+
+```shell
+docker run --rm -p 7860:7860 radtts-vocos-api
+```

example.py

@@ -4,7 +4,8 @@ pip3 install gradio_client
 
 from gradio_client import Client
 
-client = Client("Yehor/radtts-uk-vocos-demo")
+# client = Client("Yehor/radtts-uk-vocos-demo")
+client = Client("http://0.0.0.0:7860/")
 
 audio, metrics = client.predict(
     param_0="Ви можете протестувати синтез мовлення українською мовою. Просто введіть текст, який ви хочете прослухати.",  # text
@@ -20,7 +21,8 @@ audio, metrics = client.predict(
     param_10=0.666,  # sigma_token_duration
     param_11=1,  # sigma_f0
     param_12=1,  # sigma_energy
-    api_name="/inference_zerogpu",
+    # api_name="/inference_zerogpu",
+    api_name="/inference_cpu",
 )
 
 filename = audio["value"]