Update Dockerfile

Dockerfile

@@ -1,9 +1,12 @@
 # Start with a base image that includes CUDA
-FROM nvidia/cuda:11.4.1-base-ubuntu20.04
+# FROM nvidia/cuda:11.4.1-base-ubuntu20.04
+FROM huggingface/transformers-tensorflow-gpu
 
 RUN apt update && apt install -y software-properties-common
 RUN add-apt-repository ppa:deadsnakes/ppa
-RUN apt install -y python3.9
+# RUN apt install -y python3.
+
+RUN python --version
 
 # Set the timezone
 ENV TZ=Europe/Berlin
@@ -17,12 +20,13 @@ RUN dpkg-reconfigure --frontend noninteractive tzdata
 # Install the libglib2.0-0 package
 RUN apt-get update && apt-get install -y libgl1-mesa-glx
 RUN apt-get update && apt-get install -y libglib2.0-0
-RUN apt-get update && apt-get install -y python3.9 python3-pip
+RUN apt-get update && apt-get install -y python3 python3-pip
 
 # Install the Python dependencies
 WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
-RUN  python3.9 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN  python3 -m pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -45,4 +49,4 @@ RUN chmod o+w /home/user/app
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . /home/user/app
 
-CMD ["python3.9", "main.py"]
+CMD ["python3", "main.py"]