Spaces:
Running
Running
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -30,12 +30,12 @@ ENV PORT=7860 \
|
|
30 |
WORKDIR ${HOME}/mateo
|
31 |
|
32 |
# Clone the repository
|
33 |
-
RUN git clone --depth 1 --branch v1.
|
34 |
WORKDIR mateo-demo
|
35 |
|
36 |
# Install Python dependencies with conditional torch installation
|
37 |
RUN python -m pip install --no-cache-dir --upgrade pip wheel setuptools \
|
38 |
-
&& python -m pip install --no-cache-dir torch==2.
|
39 |
&& python -m pip install --no-cache-dir --upgrade .
|
40 |
|
41 |
# Pre-download default models
|
|
|
30 |
WORKDIR ${HOME}/mateo
|
31 |
|
32 |
# Clone the repository
|
33 |
+
RUN git clone --depth 1 --branch v1.5.0 https://github.com/BramVanroy/mateo-demo.git
|
34 |
WORKDIR mateo-demo
|
35 |
|
36 |
# Install Python dependencies with conditional torch installation
|
37 |
RUN python -m pip install --no-cache-dir --upgrade pip wheel setuptools \
|
38 |
+
&& python -m pip install --no-cache-dir torch==2.6.0+cpu -f https://download.pytorch.org/whl/torch \
|
39 |
&& python -m pip install --no-cache-dir --upgrade .
|
40 |
|
41 |
# Pre-download default models
|