d0tpy commited on
Commit
8b078b7
·
verified ·
1 Parent(s): 61ee4c8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -0
Dockerfile CHANGED
@@ -8,6 +8,9 @@ COPY . /app
8
  RUN pip install --no-cache-dir -r requirements.txt
9
  RUN apt-get update && apt-get install -y libgl1-mesa-glx
10
 
 
 
 
11
  # Make port 80 available to the world outside this container
12
  EXPOSE 7860
13
 
 
8
  RUN pip install --no-cache-dir -r requirements.txt
9
  RUN apt-get update && apt-get install -y libgl1-mesa-glx
10
 
11
+ # Command to modify the file before app startup
12
+ RUN sed -i 's/from torchvision.transforms.functional_tensor import rgb_to_grayscale/from torchvision.transforms.functional import rgb_to_grayscale/' /usr/local/lib/python3.9/site-packages/basicsr/data/degradations.py
13
+
14
  # Make port 80 available to the world outside this container
15
  EXPOSE 7860
16