Update Dockerfile

Dockerfile

@@ -1,9 +1,12 @@
 FROM nvidia/cuda:12.2.0-cudnn8-runtime-ubuntu22.04
 
-RUN apt update && apt install -y python3-pip git
-RUN pip3 install torch transformers fastapi uvicorn pillow
+RUN apt-get update && apt-get install -y python3 python3-pip git && \
+    pip3 install --upgrade pip
 
-COPY . /app
+COPY requirements.txt /app/requirements.txt
 WORKDIR /app
+RUN pip3 install -r requirements.txt
+
+COPY . /app
 
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "5000"]