HeliosZhao commited on
Commit
47b5f74
1 Parent(s): 6e03515

update dockerfile

Browse files
Files changed (2) hide show
  1. Dockerfile +59 -0
  2. app.py +2 -2
Dockerfile ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
2
+ ENV DEBIAN_FRONTEND=noninteractive
3
+ RUN apt-get update && \
4
+ apt-get upgrade -y && \
5
+ apt-get install -y --no-install-recommends \
6
+ git \
7
+ git-lfs \
8
+ wget \
9
+ curl \
10
+ # ffmpeg \
11
+ ffmpeg \
12
+ x264 \
13
+ # python build dependencies \
14
+ build-essential \
15
+ libssl-dev \
16
+ zlib1g-dev \
17
+ libbz2-dev \
18
+ libreadline-dev \
19
+ libsqlite3-dev \
20
+ libncursesw5-dev \
21
+ xz-utils \
22
+ tk-dev \
23
+ libxml2-dev \
24
+ libxmlsec1-dev \
25
+ libffi-dev \
26
+ liblzma-dev && \
27
+ apt-get clean && \
28
+ rm -rf /var/lib/apt/lists/*
29
+
30
+ RUN useradd -m -u 1000 user
31
+ USER user
32
+ ENV HOME=/home/user \
33
+ PATH=/home/user/.local/bin:${PATH}
34
+ WORKDIR ${HOME}/app
35
+
36
+ RUN curl https://pyenv.run | bash
37
+ ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
38
+ ENV PYTHON_VERSION=3.10.9
39
+ RUN pyenv install ${PYTHON_VERSION} && \
40
+ pyenv global ${PYTHON_VERSION} && \
41
+ pyenv rehash && \
42
+ pip install --no-cache-dir -U pip setuptools wheel
43
+
44
+ RUN pip install --no-cache-dir -U torch==1.13.1 torchvision==0.14.1
45
+ COPY --chown=1000 requirements.txt /tmp/requirements.txt
46
+ RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
47
+ RUN pip install -e Make-A-Protagonist/experts/GroundedSAM/segment_anything
48
+ RUN pip install -e Make-A-Protagonist/experts/GroundedSAM/GroundingDINO
49
+
50
+ COPY --chown=1000 . ${HOME}/app
51
+ RUN cd Make-A-Protagonist && patch -p1 < ../patch
52
+ ENV PYTHONPATH=${HOME}/app \
53
+ PYTHONUNBUFFERED=1 \
54
+ GRADIO_ALLOW_FLAGGING=never \
55
+ GRADIO_NUM_PORTS=1 \
56
+ GRADIO_SERVER_NAME=0.0.0.0 \
57
+ GRADIO_THEME=huggingface \
58
+ SYSTEM=spaces
59
+ CMD ["python", "app.py"]
app.py CHANGED
@@ -6,8 +6,8 @@ import os
6
  import sys
7
  import warnings
8
 
9
- os.system("python -m pip install -e Make-A-Protagonist/experts/GroundedSAM/segment_anything")
10
- os.system("python -m pip install -e Make-A-Protagonist/experts/GroundedSAM/GroundingDINO")
11
  # os.system("pip install --upgrade diffusers[torch]")
12
  warnings.filterwarnings("ignore")
13
 
 
6
  import sys
7
  import warnings
8
 
9
+ # os.system("python -m pip install -e Make-A-Protagonist/experts/GroundedSAM/segment_anything")
10
+ # os.system("python -m pip install -e Make-A-Protagonist/experts/GroundedSAM/GroundingDINO")
11
  # os.system("pip install --upgrade diffusers[torch]")
12
  warnings.filterwarnings("ignore")
13