Update Dockerfile

Dockerfile

@@ -1,36 +1,21 @@
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu20.04
+FROM python:3.10.9
 
-# Set the environment variable to noninteractive to avoid tzdata prompts
-ENV DEBIAN_FRONTEND=noninteractive
-
-# Install Python and other dependencies
-RUN apt-get update && apt-get install -y \
-    python3.10 \
-    python3-pip \
-    ffmpeg \
-    tzdata \
-    && apt-get clean \
-    && rm -rf /var/lib/apt/lists/*
-
-# Configure the timezone (change to your preferred timezone)
-RUN ln -fs /usr/share/zoneinfo/America/New_York /etc/localtime && \
-    dpkg-reconfigure --frontend noninteractive tzdata
-
-# Set up the working directory
+# Use a more secure method for setting the working directory
 WORKDIR /app
 
-# Copy the requirements file and install dependencies
+# Copy the requirements file
 COPY ./requirements.txt /app/requirements.txt
-RUN python3 -m pip install --upgrade pip \
-    && python3 -m pip install --no-cache-dir --upgrade -r /app/requirements.txt
+
+# Install dependencies
+RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
+RUN apt update && apt install -y ffmpeg
 
 # Add and configure the user
 RUN useradd -m -u 1000 user
 
 # Set environment variables for the user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    CUDA_HOME=/
+    PATH=/home/user/.local/bin:$PATH
 
 # Set the user and working directory
 USER user
@@ -45,6 +30,7 @@ RUN chmod -R 777 $HOME/app
 # Set additional environment variables
 ENV NUMBA_CACHE_DIR=/tmp/NUMBA_CACHE_DIR/
 ENV MPLCONFIGDIR=/tmp/MPLCONFIGDIR/
+ENV HOME=/home/user
 ENV PATH="/opt/venv/bin:$PATH"
 
 # Set the command to run the application