Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +10 -8
Dockerfile
CHANGED
@@ -1,18 +1,20 @@
|
|
1 |
FROM python:3.9-slim
|
2 |
|
3 |
-
# Install system dependencies
|
4 |
RUN apt-get update && apt-get install -y \
|
5 |
-
curl \
|
6 |
build-essential \
|
7 |
-
cargo \
|
8 |
-
libsm6 \
|
9 |
-
libxext6 \
|
10 |
-
libgl1-mesa-glx \
|
11 |
ffmpeg \
|
12 |
-
git \
|
13 |
-
git-lfs \
|
14 |
&& rm -rf /var/lib/apt/lists/*
|
15 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
16 |
# Set the working directory
|
17 |
WORKDIR /app
|
18 |
|
|
|
1 |
FROM python:3.9-slim
|
2 |
|
|
|
3 |
RUN apt-get update && apt-get install -y \
|
|
|
4 |
build-essential \
|
|
|
|
|
|
|
|
|
5 |
ffmpeg \
|
|
|
|
|
6 |
&& rm -rf /var/lib/apt/lists/*
|
7 |
|
8 |
+
RUN pip install --no-cache-dir --upgrade pip && \
|
9 |
+
pip install --no-cache-dir \
|
10 |
+
torch==2.0.1 torchvision==0.15.2 \
|
11 |
+
diffusers==0.16.0 \
|
12 |
+
accelerate==0.21.0 \
|
13 |
+
transformers>=4.26.0 \
|
14 |
+
Pillow==9.4.0 \
|
15 |
+
opencv-python==4.8.0.76
|
16 |
+
|
17 |
+
|
18 |
# Set the working directory
|
19 |
WORKDIR /app
|
20 |
|