Update Dockerfile

Dockerfile

@@ -1,12 +1,19 @@
-FROM nvidia/cuda:12.2.0-cudnn8-runtime-ubuntu22.04
+FROM nvidia/cuda:11.8.0-runtime-ubuntu22.04
 
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Python + dependencies
 RUN apt-get update && apt-get install -y python3 python3-pip git && \
     pip3 install --upgrade pip
 
-COPY requirements.txt /app/requirements.txt
+# Set working dir
 WORKDIR /app
-RUN pip3 install -r requirements.txt
 
-COPY . /app
+# Copy and install requirements
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy app code
+COPY . .
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"]