Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -34,11 +34,11 @@ RUN pip install --no-cache-dir -r requirements.txt
|
|
34 |
COPY pretrained/ $HOME/app/universal-image-restoration/config/daclip-sde/options/
|
35 |
|
36 |
# Set the working directory to the app.py file's location
|
37 |
-
WORKDIR $HOME/app/universal-image-restoration/
|
38 |
|
39 |
# Set the environment variable to specify the GPU device
|
40 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
41 |
ENV CUDA_VISIBLE_DEVICES=0
|
42 |
|
43 |
# Run your app.py script
|
44 |
-
CMD ["python", "app.py"]
|
|
|
34 |
COPY pretrained/ $HOME/app/universal-image-restoration/config/daclip-sde/options/
|
35 |
|
36 |
# Set the working directory to the app.py file's location
|
37 |
+
WORKDIR $HOME/app/universal-image-restoration/
|
38 |
|
39 |
# Set the environment variable to specify the GPU device
|
40 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
41 |
ENV CUDA_VISIBLE_DEVICES=0
|
42 |
|
43 |
# Run your app.py script
|
44 |
+
CMD ["python", "/config/daclip-sde/app.py"]
|