Update Dockerfile

Dockerfile

@@ -1,20 +1,27 @@
-FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
+FROM nvidia/cuda:12.1.1-cudnn9-runtime-ubuntu22.04
 
-# Install system dependencies
+# Install system deps
 RUN apt-get update && apt-get install -y \
     python3.10 python3-pip git ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
+COPY . .
 
-# Copy requirements and install
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Force clean onnxruntime-gpu install
-RUN pip uninstall -y onnxruntime onnxruntime-gpu && \
-    pip install onnxruntime-gpu
+RUN pip install --no-cache-dir \
+    gradio \
+    insightface==0.7.3 \
+    onnxruntime-gpu==1.18.0 \
+    opencv-python-headless \
+    numpy
 
-COPY . .
+# Pre-download models
+RUN python -c "
+import insightface
+from insightface.app import FaceAnalysis
+app = FaceAnalysis(name='buffalo_l', providers=['CUDAExecutionProvider'])
+app.prepare(ctx_id=0, det_size=(640, 640))
+print('Models ready')
+"
 
-CMD ["python3", "app.py"]
+CMD ["python", "app.py"]