Update Dockerfile

Dockerfile

@@ -1,33 +1,17 @@
 # Use NVIDIA CUDA base image with Ubuntu 22.04, CUDA 12.2, and cuDNN 8
 FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
 
-# Set Python to run in unbuffered mode for better logging
-ENV PYTHONUNBUFFERED=1
+USER root
 
-# Install Python and system dependencies in a single layer
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    python3.10 \
-    python3-pip \
-    libgl1 \
-    libglib2.0-0 \
-    libsm6 \
-    libxext6 \
-    libxrender-dev \
-    && ln -s /usr/bin/python3.10 /usr/bin/python \
-    && rm -rf /var/lib/apt/lists/*
+RUN chmod 1777 /tmp \
+    && apt update -q && apt install -y ca-certificates wget libgl1 \
+    && 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
 
-# Copy and install Python requirements
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install fastapi[all] 
 
-# Copy application code
-COPY main.py utils.py ./
 
-# Verify the application can be imported without errors
-RUN python -c "from main import app"
-
-# Expose the application port
-EXPOSE 7860
-
-# Start the FastAPI server
+COPY main.py main.py
+RUN python main.py
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]