More fixes

Dockerfile

@@ -1,8 +1,10 @@
 FROM nvidia/cuda:12.4.1-cudnn8-devel-ubuntu22.04
 
-# Install Python and required system libraries
+# Install Python 3.10 and required system libraries
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.10 \
+    python3.10-venv \
+    python3.10-dev \
     python3-pip \
     build-essential \
     gcc \
@@ -14,8 +16,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libxrender-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Create Python 3.10 symlink
-RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1
+# Set Python 3.10 as default
+RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 && \
+    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
 
 WORKDIR /code
 

main.py

@@ -8,11 +8,14 @@ import json
 import os
 from datetime import datetime
 from handler import EndpointHandler
-from debug import run_all_checks
 import numpy as np
 
-# Run diagnostics on startup
-run_all_checks()
+# Run diagnostics on startup (safe import)
+try:
+    from debug import run_all_checks
+    run_all_checks()
+except Exception as e:
+    print(f"⚠️  Diagnostics failed to run: {e}")
 
 app = FastAPI()
 

requirements.txt

@@ -1,10 +1,10 @@
 azure-storage-blob
-onnxruntime-gpu==1.18.0
+onnxruntime-gpu==1.16.3
 insightface
 opencv-python-headless
 flask
 flask-cors
-numpy
+numpy<2.0
 scipy
 pillow
 requests