Update Dockerfile

Dockerfile

@@ -5,7 +5,10 @@ FROM python:3.9-slim
 ENV PYTHONDONTWRITEBYTECODE 1
 ENV PYTHONUNBUFFERED 1
 
-# Step 3: Install system dependencies
+# Step 3: Set the environment variable for PyTorch cache
+ENV TORCH_HOME=/app/.cache
+
+# Step 4: Install system dependencies
 RUN apt-get update && apt-get install -y \
     gcc \
     g++ \
@@ -14,20 +17,20 @@ RUN apt-get update && apt-get install -y \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Step 4: Set the working directory inside the container
+# Step 5: Set the working directory inside the container
 WORKDIR /app
 
-# Step 5: Copy the requirements file into the container
+# Step 6: Copy the requirements file into the container
 COPY requirements.txt /app/
 
-# Step 6: Install the Python dependencies
+# Step 7: Install the Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Step 7: Copy the rest of the application files into the container
+# Step 8: Copy the rest of the application files into the container
 COPY . /app/
 
-# Step 8: Expose the port that the Flask app will run on
+# Step 9: Expose the port that the Flask app will run on
 EXPOSE 5000
 
-# Step 9: Set the entry point to start the Flask application
+# Step 10: Set the entry point to start the Flask application
 CMD ["python", "app.py"]