MogensR commited on
Commit
dc3baf7
·
1 Parent(s): a7bd8de

Update docker/Dockerfile.cpu

Browse files
Files changed (1) hide show
  1. docker/Dockerfile.cpu +9 -6
docker/Dockerfile.cpu CHANGED
@@ -26,18 +26,20 @@ WORKDIR /app
26
  COPY requirements-cpu.txt requirements.txt ./
27
  RUN pip install --no-cache-dir --upgrade pip && \
28
  pip install --no-cache-dir \
29
- torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu \
30
- torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cpu && \
31
- pip install --no-cache-dir -r requirements.txt
 
 
32
 
33
  # Create app user
34
  RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
35
  USER appuser
36
 
37
- # Copy application
38
  COPY --chown=appuser:appuser . .
39
 
40
- # Create directories
41
  RUN mkdir -p logs uploads outputs models/.cache
42
 
43
  # Health check
@@ -46,4 +48,5 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
46
 
47
  EXPOSE 7860 8000
48
 
49
- CMD ["python", "-u", "app.py"]
 
 
26
  COPY requirements-cpu.txt requirements.txt ./
27
  RUN pip install --no-cache-dir --upgrade pip && \
28
  pip install --no-cache-dir \
29
+ torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu \
30
+ torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cpu && \
31
+ pip install --no-cache-dir -r requirements.txt && \
32
+ pip install --no-deps \
33
+ git+https://github.com/pq-yang/MatAnyone.git@2234ce5cdc487749515518bd035b5e18bccea3da
34
 
35
  # Create app user
36
  RUN useradd -m -u 1000 appuser && chown -R appuser:appuser /app
37
  USER appuser
38
 
39
+ # Copy application code
40
  COPY --chown=appuser:appuser . .
41
 
42
+ # Create runtime directories
43
  RUN mkdir -p logs uploads outputs models/.cache
44
 
45
  # Health check
 
48
 
49
  EXPOSE 7860 8000
50
 
51
+ # Launch
52
+ CMD ["python", "-u", "app.py"]