Update Dockerfile

Dockerfile

@@ -19,14 +19,15 @@ WORKDIR /app
 
 # Install Python dependencies
 RUN pip install --upgrade pip
-RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-RUN pip install transformers accelerate requests psutil termcolor sk-learn
 
 # Copy the current directory contents into the container at /app
 COPY . /app
 
+# install requirements
+RUN pip instal -r requirements.txt
+
 # Unzip the data.zip file into the /app/data directory
 RUN unzip data.zip -d /app/data
 
 # Set the default command to run when starting the container
-CMD ["python", "m3.py"]
+CMD ["python", "m5.py"]