sandbox338 commited on
Commit
7d0b950
·
verified ·
1 Parent(s): a6ec678

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -32,8 +32,9 @@ COPY src/ ./src/
32
  # Install Python dependencies
33
  RUN pip install --upgrade pip && \
34
  pip install "numpy<2.0" && \
35
- pip install -r requirements.txt
36
-
 
37
  # Create writable directories
38
  RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib /tmp && \
39
  chmod -R 777 /app /tmp
 
32
  # Install Python dependencies
33
  RUN pip install --upgrade pip && \
34
  pip install "numpy<2.0" && \
35
+ pip install -r requirements.txt && \
36
+ pip install torchvision==0.15.2 && \
37
+ pip install 'git+https://github.com/facebookresearch/detectron2.git'
38
  # Create writable directories
39
  RUN mkdir -p /app/.torch /app/.streamlit /app/.config/matplotlib /tmp && \
40
  chmod -R 777 /app /tmp