Update Dockerfile

Dockerfile

@@ -1,4 +1,5 @@
-FROM python:3.10.9
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
+# FROM python:3.10.9
 
 # Use a more secure method for setting the working directory
 WORKDIR /app
@@ -7,7 +8,8 @@ WORKDIR /app
 COPY ./requirements.txt /app/requirements.txt
 
 # Install dependencies
-RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
+RUN apt update && apt install -y python3 python3-pip
+RUN python3 -m pip install --no-cache-dir --upgrade -r /app/requirements.txt
 RUN apt update && apt install -y ffmpeg
 
 # Add and configure the user