Update Dockerfile

Dockerfile

@@ -1,18 +1,14 @@
-FROM python:3.9
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
 
-# Create non-root user
-RUN useradd -m -u 1000 user
-USER user
+# System setup
+RUN apt-get update && apt-get install -y python3-pip git libgl1
 
-# Set up path and working directory
-ENV PATH="/home/user/.local/bin:$PATH"
-WORKDIR /app
-
-# Copy files
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+# Python dependencies
+COPY requirements.txt .
+RUN pip3 install --upgrade pip && pip3 install -r requirements.txt
 
-COPY --chown=user . /app
+# App
+COPY . /app
+WORKDIR /app
 
-# Start server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]