Testing: Dockerfile image on going for OOM

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.10.0-slim-buster
+FROM python:3.10.9
 
 WORKDIR /app
 
@@ -12,9 +12,7 @@ COPY .dockerignore .
 COPY . .
 
 RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y
-RUN export PYTORCH_ENABLE_MPS_FALLBACK=1
 
 ENV SECRET_KEY_FILE=/run/secrets/secret_key.json
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--log-level", "error"]
-# CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--log-level", "trace", "--ssl-keyfile=./app/keys/private.key", "--ssl-certfile=./app/keys/certificate.crt", "--ws-max-size", "167772160", "--ws-max-queue", "128"]
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--log-level", "trace", "--ssl-keyfile", "./app/keys/private.key", "--ssl-certfile", "./app/keys/certificate.crt"]

app/main.py

@@ -16,7 +16,7 @@
 EXTRACTOR_THRESHOLD = 0.85
 
 app = FastAPI()
-extractor = SkeletonExtractor(pretrained_bool=True, number_of_keypoints=17, device='mps')
+extractor = SkeletonExtractor(pretrained_bool=True, number_of_keypoints=17, device='cpu')
 preprocessor = DataPreprocessing()
 metrics = Metrics()
 

models.py

@@ -156,7 +156,7 @@ def extract(self, video_tensor: cv2.VideoCapture, score_threshold: float = 0.93,
             total_fps += fps
             frame_count += 1
 
-            pbar.set_postfix({"FPS": f"{fps:.2f}", "Average FPS": f"{total_fps / frame_count:.2f}"})
+            pbar.set_postfix({"FPS": f"{fps:.2f}", "Average FPS": f"{total_fps / frame_count:.2f}, Skeleton shapes: {len(extracted_skeletons)}"})
             pbar.update(1)
 
         pbar.close()