Update Dockerfile

Dockerfile

@@ -1,5 +1,9 @@
-# Use the official Python image
-FROM python:3.9-slim
+# Use the official NVIDIA CUDA image
+FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu20.04
+
+# Set environment variables for non-interactive installation and timezone
+ENV DEBIAN_FRONTEND=noninteractive \
+    TZ=Europe/Paris
 
 # Set the working directory
 WORKDIR /app
@@ -7,27 +11,29 @@ WORKDIR /app
 # Create a non-root user
 RUN useradd -m appuser
 
+# Install system dependencies and Python
+RUN apt-get update && apt-get install -y \
+    python3-pip \
+    python3-dev \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install any necessary Python packages (e.g., Flask, transformers, torch, etc.)
+COPY requirements.txt requirements.txt
+RUN pip3 install --no-cache-dir -r requirements.txt
+
 # Create writable directories for Hugging Face and Triton cache
 RUN mkdir -p /workspace/huggingface_cache /home/appuser/.triton && \
     chmod -R 777 /workspace/huggingface_cache /home/appuser/.triton
 
-# Set environment variables
+# Set environment variables for Hugging Face and Triton cache
 ENV HF_HOME=/workspace/huggingface_cache
 ENV TRITON_CACHE_DIR=/home/appuser/.triton
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
-
 # Set umask to ensure proper file permissions
 RUN umask 002
 
-# Copy the requirements file into the image
-COPY requirements.txt requirements.txt
-
-# Install required packages
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy the rest of the application code
+# Copy the application code into the image
 COPY . .
 
 # Change ownership to non-root user
@@ -40,4 +46,4 @@ USER appuser
 EXPOSE 7860
 
 # Run the application
-CMD ["python", "app.py"]
+CMD ["python3", "app.py"]