Vo Minh Vu commited on
Commit
7959dbd
·
1 Parent(s): 42a5e65

Add Dockerfile and update requirements for proper build

Browse files
Files changed (2) hide show
  1. Dockerfile +29 -0
  2. requirements.txt +7 -5
Dockerfile ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM python:3.10
2
+
3
+ # Install system dependencies
4
+ RUN apt-get update && apt-get install -y \
5
+ git \
6
+ git-lfs \
7
+ ffmpeg \
8
+ libsm6 \
9
+ libxext6 \
10
+ cmake \
11
+ rsync \
12
+ libgl1-mesa-glx \
13
+ && rm -rf /var/lib/apt/lists/* \
14
+ && git lfs install
15
+
16
+ # Set working directory
17
+ WORKDIR /app
18
+
19
+ # Copy all necessary files
20
+ COPY . .
21
+
22
+ # Install Python dependencies
23
+ RUN pip install --no-cache-dir -r requirements.txt
24
+
25
+ # Set environment variables
26
+ ENV PYTHONPATH=/app
27
+
28
+ # Run the application
29
+ CMD ["python", "app.py"]
requirements.txt CHANGED
@@ -1,14 +1,18 @@
1
- -e .
2
  torch>=2.0.0
3
  torchvision>=0.15.0
4
  torchaudio>=2.0.0
 
 
 
 
 
 
5
  einops>=0.6.0
6
  jaxtyping>=0.2.0
7
  omegaconf>=2.3.0
8
- transformers>=4.30.0
9
  open_clip_torch>=2.0.0
10
  trimesh>=3.9.0
11
- numpy>=1.24.0
12
  huggingface-hub>=0.15.0
13
  rembg>=2.0.0
14
  onnxruntime>=1.14.0
@@ -16,7 +20,5 @@ pynanoinstantmeshes>=0.0.3
16
  scipy>=1.10.0
17
  scikit-image>=0.20.0
18
  opencv-python-headless>=4.7.0
19
- gradio>=3.50.2
20
- Pillow>=9.0.0
21
  pathlib>=1.0.1
22
  gpytoolbox
 
1
+ # Core dependencies
2
  torch>=2.0.0
3
  torchvision>=0.15.0
4
  torchaudio>=2.0.0
5
+ transformers>=4.30.0
6
+ gradio>=3.50.2
7
+ pillow>=9.0.0
8
+ numpy>=1.24.0
9
+
10
+ # Additional dependencies
11
  einops>=0.6.0
12
  jaxtyping>=0.2.0
13
  omegaconf>=2.3.0
 
14
  open_clip_torch>=2.0.0
15
  trimesh>=3.9.0
 
16
  huggingface-hub>=0.15.0
17
  rembg>=2.0.0
18
  onnxruntime>=1.14.0
 
20
  scipy>=1.10.0
21
  scikit-image>=0.20.0
22
  opencv-python-headless>=4.7.0
 
 
23
  pathlib>=1.0.1
24
  gpytoolbox