John6666 commited on
Commit
1f03703
1 Parent(s): 3f4356b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +34 -34
Dockerfile CHANGED
@@ -1,35 +1,35 @@
1
- FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
2
-
3
- # Install dependencies
4
- RUN apt-get update && apt-get install -y \
5
- git \
6
- wget \
7
- unzip \
8
- vim \
9
- ffmpeg \
10
- libsm6 \
11
- libxext6
12
-
13
- RUN apt-get clean && rm -rf /var/lib/apt/lists/*
14
-
15
- RUN useradd -m -u 1000 user
16
-
17
- USER user
18
-
19
- ENV HOME=/home/user \
20
- PATH=/home/user/.local/bin:$PATH
21
-
22
- WORKDIR $HOME/mapper
23
-
24
- RUN pip install --no-cache-dir gradio[oauth]==4.36.1 "uvicorn>=0.14.0" spaces
25
-
26
- COPY --chown=user . $HOME/mapper
27
-
28
- # Install Python dependencies
29
- RUN pip install --no-cache-dir -r requirements.txt
30
-
31
- # Get Weights
32
- RUN bash get_weights.sh
33
-
34
- # Start the app
35
  CMD ["python", "app.py"]
 
1
+ FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
2
+
3
+ # Install dependencies
4
+ RUN apt-get update && apt-get install -y \
5
+ git \
6
+ wget \
7
+ unzip \
8
+ vim \
9
+ ffmpeg \
10
+ libsm6 \
11
+ libxext6
12
+
13
+ RUN apt-get clean && rm -rf /var/lib/apt/lists/*
14
+
15
+ RUN useradd -m -u 1000 user
16
+
17
+ USER user
18
+
19
+ ENV HOME=/home/user \
20
+ PATH=/home/user/.local/bin:$PATH
21
+
22
+ WORKDIR $HOME/mapper
23
+
24
+ RUN pip install --no-cache-dir "uvicorn>=0.14.0" spaces
25
+
26
+ COPY --chown=user . $HOME/mapper
27
+
28
+ # Install Python dependencies
29
+ RUN pip install --no-cache-dir -r requirements.txt
30
+
31
+ # Get Weights
32
+ RUN bash get_weights.sh
33
+
34
+ # Start the app
35
  CMD ["python", "app.py"]