fffiloni commited on
Commit
73b9ac7
·
verified ·
1 Parent(s): 0a57c89

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -5
Dockerfile CHANGED
@@ -43,8 +43,7 @@ ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
43
  ENV CUDA_VISIBLE_DEVICES=0
44
 
45
 
46
- # Set the working directory to the user's home directory
47
- WORKDIR $HOME/app
48
 
49
  # Install system dependencies as root
50
  USER root
@@ -54,13 +53,16 @@ RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
54
  RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
55
 
56
  # Clone the repository (adjust the URL if needed)
57
- RUN git clone --recursive https://github.com/jnjaby/KEEP.git $HOME/app
58
 
59
  # Set Git ownership for the cloned repository
60
  USER root
61
  RUN git config --global --add safe.directory $HOME/app
62
  USER user
63
 
 
 
 
64
  # Copy the app.py script into the container
65
  COPY app.py .
66
  COPY requirements_HF.txt .
@@ -74,8 +76,7 @@ RUN pip install cupy==10.4.0
74
 
75
  USER root
76
  # Install basicsr
77
- RUN cd $HOME/app && \
78
- if [ -d "basicsr" ]; then \
79
  cd basicsr && \
80
  if [ ! -f "basicsr/VERSION" ]; then \
81
  echo "1.3.2" > basicsr/VERSION; \
 
43
  ENV CUDA_VISIBLE_DEVICES=0
44
 
45
 
46
+
 
47
 
48
  # Install system dependencies as root
49
  USER root
 
53
  RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
54
 
55
  # Clone the repository (adjust the URL if needed)
56
+ RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
57
 
58
  # Set Git ownership for the cloned repository
59
  USER root
60
  RUN git config --global --add safe.directory $HOME/app
61
  USER user
62
 
63
+ # Set the working directory to the user's home directory
64
+ WORKDIR $HOME/app
65
+
66
  # Copy the app.py script into the container
67
  COPY app.py .
68
  COPY requirements_HF.txt .
 
76
 
77
  USER root
78
  # Install basicsr
79
+ RUN if [ -d "basicsr" ]; then \
 
80
  cd basicsr && \
81
  if [ ! -f "basicsr/VERSION" ]; then \
82
  echo "1.3.2" > basicsr/VERSION; \