fffiloni commited on
Commit
cb26105
β€’
1 Parent(s): fe10180

adapt Dockerfile for new feature

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -32,7 +32,7 @@ RUN git clone https://github.com/sczhou/ProPainter.git $HOME/app
32
  RUN pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
33
 
34
  # Install dependencies
35
- RUN pip install --no-cache-dir -r requirements.txt gradio==3.48.0
36
 
37
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/i3d_rgb_imagenet.pt -d $HOME/app/weigths -o i3d_rgb_imagenet.pt
38
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/raft-things.pth -d $HOME/app/weights -o raft-things.pth
 
32
  RUN pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
33
 
34
  # Install dependencies
35
+ RUN pip install --no-cache-dir -r requirements.txt gradio==3.48.0 opencv-python moviepy gradio_client
36
 
37
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/i3d_rgb_imagenet.pt -d $HOME/app/weigths -o i3d_rgb_imagenet.pt
38
  RUN aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/camenduru/ProPainter/resolve/main/raft-things.pth -d $HOME/app/weights -o raft-things.pth