Yehor commited on
Commit
a27324a
·
1 Parent(s): 4bab821

Add a Dockerfile

Browse files
Files changed (4) hide show
  1. .dockerignore +6 -0
  2. Dockerfile +59 -0
  3. README.md +12 -0
  4. example.py +4 -2
.dockerignore ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ .idea/
2
+ .venv/
3
+ .ruff_cache/
4
+ __pycache__/
5
+
6
+ models/
Dockerfile ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
2
+
3
+ ENV DEBIAN_FRONTEND=noninteractive
4
+
5
+ RUN apt-get update && \
6
+ apt-get upgrade -y && \
7
+ apt-get install -y --no-install-recommends \
8
+ git \
9
+ git-lfs \
10
+ wget \
11
+ curl \
12
+ # python build dependencies \
13
+ build-essential \
14
+ libssl-dev \
15
+ zlib1g-dev \
16
+ libbz2-dev \
17
+ libreadline-dev \
18
+ libsqlite3-dev \
19
+ libncursesw5-dev \
20
+ xz-utils \
21
+ tk-dev \
22
+ libxml2-dev \
23
+ libxmlsec1-dev \
24
+ libffi-dev \
25
+ liblzma-dev \
26
+ # gradio dependencies \
27
+ ffmpeg \
28
+ && apt-get clean \
29
+ && rm -rf /var/lib/apt/lists/*
30
+
31
+
32
+ RUN useradd -m -u 1000 user
33
+ USER user
34
+ ENV HOME=/home/user \
35
+ PATH=/home/user/.local/bin:${PATH}
36
+ WORKDIR ${HOME}/app
37
+
38
+ RUN curl https://pyenv.run | bash
39
+ ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
40
+ ARG PYTHON_VERSION=3.10.12
41
+ RUN pyenv install ${PYTHON_VERSION} && \
42
+ pyenv global ${PYTHON_VERSION} && \
43
+ pyenv rehash && \
44
+ pip install --no-cache-dir -U pip setuptools wheel && \
45
+ pip install packaging ninja
46
+
47
+ COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
48
+ RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt
49
+
50
+ COPY --chown=1000 . ${HOME}/app
51
+ ENV PYTHONPATH=${HOME}/app \
52
+ PYTHONUNBUFFERED=1 \
53
+ GRADIO_ALLOW_FLAGGING=never \
54
+ GRADIO_NUM_PORTS=1 \
55
+ GRADIO_SERVER_NAME=0.0.0.0 \
56
+ GRADIO_THEME=huggingface \
57
+ SYSTEM=spaces
58
+
59
+ CMD ["python", "app.py"]
README.md CHANGED
@@ -24,3 +24,15 @@ uv pip install -r requirements.txt
24
  ```shell
25
  python app.py
26
  ```
 
 
 
 
 
 
 
 
 
 
 
 
 
24
  ```shell
25
  python app.py
26
  ```
27
+
28
+ ## Build Docker image
29
+
30
+ ```shell
31
+ docker build -t radtts-vocos-api .
32
+ ```
33
+
34
+ ## Run a container
35
+
36
+ ```shell
37
+ docker run --rm -p 7860:7860 radtts-vocos-api
38
+ ```
example.py CHANGED
@@ -4,7 +4,8 @@ pip3 install gradio_client
4
 
5
  from gradio_client import Client
6
 
7
- client = Client("Yehor/radtts-uk-vocos-demo")
 
8
 
9
  audio, metrics = client.predict(
10
  param_0="Ви можете протестувати синтез мовлення українською мовою. Просто введіть текст, який ви хочете прослухати.", # text
@@ -20,7 +21,8 @@ audio, metrics = client.predict(
20
  param_10=0.666, # sigma_token_duration
21
  param_11=1, # sigma_f0
22
  param_12=1, # sigma_energy
23
- api_name="/inference_zerogpu",
 
24
  )
25
 
26
  filename = audio["value"]
 
4
 
5
  from gradio_client import Client
6
 
7
+ # client = Client("Yehor/radtts-uk-vocos-demo")
8
+ client = Client("http://0.0.0.0:7860/")
9
 
10
  audio, metrics = client.predict(
11
  param_0="Ви можете протестувати синтез мовлення українською мовою. Просто введіть текст, який ви хочете прослухати.", # text
 
21
  param_10=0.666, # sigma_token_duration
22
  param_11=1, # sigma_f0
23
  param_12=1, # sigma_energy
24
+ # api_name="/inference_zerogpu",
25
+ api_name="/inference_cpu",
26
  )
27
 
28
  filename = audio["value"]