Update Dockerfile

Dockerfile

@@ -1,11 +1,21 @@
+# Use an official Python runtime as a parent image
 FROM python:3.9
 
+# Set the working directory to /code
 WORKDIR /code
 
-COPY ./requirements.txt /code/requirements.txt
+# Copy the current directory contents into the container at /code
+COPY . .
 
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Install any needed packages specified in requirements.txt
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    cuda-cudart-11-4=11.4.1-1 \
+    cuda-compat-11-4 \
+    && rm -rf /var/lib/apt/lists/*
 
-COPY . .
+# Install required Python packages
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-CMD ["python", "caption_api.py", "--host", "0.0.0.0", "--port", "7860"]
+# Run app.py when the container launches
+CMD ["python", "caption_api.py", "--host", "0.0.0.0", "--port", "7860"]