Update Dockerfile

Dockerfile

@@ -1,6 +1,9 @@
 # Use the NVIDIA CUDA image as the base
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu20.04
 
+# Set non-interactive mode for apt-get
+ENV DEBIAN_FRONTEND=noninteractive
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
     python3 \
@@ -36,5 +39,8 @@ RUN chown -R user:user /app
 # Switch to the "user" user
 USER user
 
+# Set the PATH to include the user's local bin directory
+ENV PATH="/home/user/.local/bin:$PATH"
+
 # Run the application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]