fffiloni commited on
Commit
ca77f10
1 Parent(s): a79cc2a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -24,10 +24,9 @@ ENV HOME=/home/user \
24
  # Set the working directory to the user's home directory
25
  WORKDIR $HOME/app
26
 
27
- COPY app.py .
28
 
29
  # Clone your repository or add your code to the container
30
- RUN git clone -b main https://github.com/sczhou/ProPainter $HOME/app
31
 
32
  # Install specific versions of PyTorch and TorchVision
33
  RUN pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
@@ -40,6 +39,8 @@ RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co
40
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/recurrent_flow_completion.pth -d $HOME/app/weights -o recurrent_flow_completion.pth
41
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/ProPainter.pth -d $HOME/app/weights -o ProPainter.pth
42
 
 
 
43
  RUN find $HOME/app
44
 
45
  # Set the environment variable to specify the GPU device
 
24
  # Set the working directory to the user's home directory
25
  WORKDIR $HOME/app
26
 
 
27
 
28
  # Clone your repository or add your code to the container
29
+ RUN git clone https://github.com/sczhou/ProPainter.git $HOME/app
30
 
31
  # Install specific versions of PyTorch and TorchVision
32
  RUN pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 
39
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/recurrent_flow_completion.pth -d $HOME/app/weights -o recurrent_flow_completion.pth
40
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/ProPainter.pth -d $HOME/app/weights -o ProPainter.pth
41
 
42
+ COPY app.py .
43
+
44
  RUN find $HOME/app
45
 
46
  # Set the environment variable to specify the GPU device