Update Dockerfile

Dockerfile

@@ -1,12 +1,5 @@
-# Use an official CUDA image as the base image
-FROM nvidia/cuda:11.4.1-cudnn8-runtime-ubuntu20.04
-
-# Install Python and pip
-RUN apt-get update && \
-    apt-get install -y python3.9 python3-pip
-
-# Install NumPy and GRPCIO
-RUN pip3 install --no-cache-dir --upgrade numpy grpcio
+# Use Hugging Face's transformers image as the base image
+FROM transformers:4.11.3-gpu
 
 # Set the working directory in the container
 WORKDIR /code
@@ -14,8 +7,8 @@ WORKDIR /code
 # Copy the requirements file
 COPY ./requirements.txt /code/requirements.txt
 
-# Install dependencies
-RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
+# Install additional dependencies
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 # Copy the application files to the container
 COPY . .