Update Dockerfile

Dockerfile

@@ -1,28 +1,32 @@
-# Use an official Python runtime as a parent image
-FROM python:3.9
+# Use NVIDIA's CUDA base image with Python
+FROM nvidia/cuda:11.8.0-runtime-ubuntu20.04
 
-# Install required system dependencies
+# Set environment variables
+ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
+ENV PATH="/usr/local/bin:$PATH"
+
+# Install Python and other dependencies
 RUN apt-get update && apt-get install -y \
+    python3 \
+    python3-pip \
+    python3-dev \
     libgl1-mesa-glx \
-    && rm -rf /var/lib/apt/lists/*
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user with a home directory
-RUN useradd -m -u 1000 user
+# Create a non-root user
+RUN useradd -m user
 
 # Set the user to the non-root user
 USER user
 
-# Add the user's local bin to the PATH
-ENV PATH="/home/user/.local/bin:$PATH"
-
 # Set the working directory to /app
 WORKDIR /app
 
-# Copy the requirements.txt file and install any dependencies
-COPY --chown=user ./requirements.txt /app/requirements.txt
-RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
+# Copy the requirements.txt file and install dependencies
+COPY --chown=user requirements.txt /app/requirements.txt
+RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
 
-# Copy the current directory contents into the container at /app
+# Copy the application code into the container
 COPY --chown=user . /app
 
 # Create the folders for uploads and results with appropriate permissions