Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -18,6 +18,7 @@ COPY --link --chown=1000 ./requirements.txt /home/user/app/requirements.txt
|
|
| 18 |
WORKDIR /home/user/app
|
| 19 |
|
| 20 |
RUN pip install -U setuptools wheel
|
|
|
|
| 21 |
RUN pip install -r requirements.txt
|
| 22 |
|
| 23 |
COPY --link --chown=1000 ./app.py /home/user/app/app.py
|
|
|
|
| 18 |
WORKDIR /home/user/app
|
| 19 |
|
| 20 |
RUN pip install -U setuptools wheel
|
| 21 |
+
RUN pip install torch==2.5.1+cu124 --index-url https://download.pytorch.org/whl/cu124
|
| 22 |
RUN pip install -r requirements.txt
|
| 23 |
|
| 24 |
COPY --link --chown=1000 ./app.py /home/user/app/app.py
|