Update Dockerfile

Dockerfile

@@ -1,18 +1,21 @@
-FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
+FROM docker.io/nvidia/cuda:12.2.2-runtime-ubuntu22.04@sha256:94c1577b2cd9dd6c0312dc04dff9cb2fdce2b268018abc3d7c2dbcacf1155000
 
-ENV DEBIAN_FRONTEND=noninteractive
+USER root
 
+# Install system dependencies
 RUN chmod 1777 /tmp \
-    && apt update -q && apt install -y ca-certificates wget libgl1 \
+    && apt update -q && apt install -y \
+    ca-certificates \
+    wget \
+    libgl1 \
+    python3-pip \  # Add Python and pip
     && wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
     && dpkg -i /tmp/cuda-keyring.deb && apt update -q \
     && apt install -y --no-install-recommends libcudnn8 libcublas-12-2
 
+# Install Python packages
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip3 install --no-cache-dir -r requirements.txt  # Use pip3
 
-COPY . .
-
-EXPOSE 7860
-
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+COPY main.py main.py
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]