Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
@@ -45,5 +45,8 @@ RUN find $HOME/app
|
|
45 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
46 |
ENV CUDA_VISIBLE_DEVICES=0
|
47 |
|
|
|
|
|
|
|
48 |
# Run your app.py script
|
49 |
-
CMD ["python", "
|
|
|
45 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
46 |
ENV CUDA_VISIBLE_DEVICES=0
|
47 |
|
48 |
+
# Set the working directory to the user's home directory
|
49 |
+
WORKDIR $HOME/app/universal-image-restoration/config/daclip-sde/
|
50 |
+
|
51 |
# Run your app.py script
|
52 |
+
CMD ["python", "app.py"]
|