Including cache

Dockerfile

@@ -1,10 +1,23 @@
 FROM python:3.10.12
 
-COPY . .
+# Create a non-root user
+RUN useradd -m -u 1000 user
+
+# Set up the working directory
+WORKDIR /app
 
-WORKDIR /
+# Copy the requirements and install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy the rest of the application
+COPY . .
 
-RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+# Create and set permissions for the cache directory
+RUN mkdir /.cache && chown -R user:user /.cache && chmod -R 777 /.cache
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+# Switch to the non-root user
+USER user
 
+# Run the application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -1,3 +1,4 @@
+import os
 from fastapi import FastAPI, UploadFile
 from fastapi.responses import JSONResponse
 from fastapi.param_functions import File
@@ -6,7 +7,6 @@ from typing import List
 import io
 from facenet_pytorch import MTCNN, InceptionResnetV1
 import torch
-import io
 from PIL import Image
 
 app = FastAPI()
@@ -19,6 +19,9 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Set the cache directory to a writable location
+os.environ['TORCH_HOME'] = '/tmp/.cache/torch'
+
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 mtcnn = MTCNN(keep_all=True, device=device)