Update Dockerfile

Dockerfile

@@ -16,7 +16,12 @@ RUN apt-get update && apt-get install -y \
 
 # Create a working directory
 WORKDIR /app
-WORKDIR /app/cache/
+
+# Create cache directory
+RUN mkdir -p /app/cache
+
+# Set the cache environment variable
+ENV TRANSFORMERS_CACHE=/app/cache
 
 # Install Python dependencies
 RUN pip install --upgrade pip
@@ -24,11 +29,11 @@ RUN pip install --upgrade pip
 # Copy the current directory contents into the container at /app
 COPY . /app
 
-# install requirements
-CMD ['pip install -r requirements.txt']
+# Install requirements
+RUN pip install -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", "m5.py"]
+CMD ["python", "m5.py"]