Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -21,6 +21,7 @@ WORKDIR /app
|
|
| 21 |
RUN pip install --upgrade pip
|
| 22 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 23 |
RUN pip install transformers accelerate
|
|
|
|
| 24 |
|
| 25 |
# Copy the current directory contents into the container at /app
|
| 26 |
COPY . /app
|
|
@@ -32,4 +33,4 @@ COPY data.zip /app/data.zip
|
|
| 32 |
RUN unzip /app/data.zip -d /app/data
|
| 33 |
|
| 34 |
# Set the default command to run when starting the container
|
| 35 |
-
CMD ["python", "
|
|
|
|
| 21 |
RUN pip install --upgrade pip
|
| 22 |
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
| 23 |
RUN pip install transformers accelerate
|
| 24 |
+
RUN pip install -r requirements.txt
|
| 25 |
|
| 26 |
# Copy the current directory contents into the container at /app
|
| 27 |
COPY . /app
|
|
|
|
| 33 |
RUN unzip /app/data.zip -d /app/data
|
| 34 |
|
| 35 |
# Set the default command to run when starting the container
|
| 36 |
+
CMD ["python", "model.py"]
|