Update to CUDA 12.4.1 to remove deprecation warning

Dockerfile

@@ -1,10 +1,21 @@
-FROM python:3.11-slim
+FROM nvidia/cuda:12.4.1-runtime-ubuntu22.04
 
-# Install system dependencies
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PYTHONUNBUFFERED=1
+
+# Install Python and system dependencies
 RUN apt-get update && apt-get install -y \
-    gcc \
-    g++ \
-    && rm -rf /var/lib/apt/lists/*
+    python3.11 \
+    python3.11-dev \
+    python3-pip \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/* \
+    && ln -s /usr/bin/python3.11 /usr/bin/python
+
+# Upgrade pip
+RUN python -m pip install --upgrade pip
 
 # Set working directory
 WORKDIR /app
@@ -22,6 +33,9 @@ COPY app/ ./app/
 RUN useradd -m -u 1000 user && chown -R user:user /app
 USER user
 
+# Set HuggingFace cache directory
+ENV HF_HOME=/tmp/huggingface
+
 # Expose port
 EXPOSE 7860