Princess3 commited on
Commit
9974267
1 Parent(s): bfeb353

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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", "your_script.py"]
 
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"]