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"]
|