Update Dockerfile

Dockerfile

@@ -1,6 +1,6 @@
 FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 
 
-# Environment variables
+# Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive \
     HF_HOME=/tmp/hf_cache \
     TRANSFORMERS_CACHE=/tmp/hf_cache \
@@ -8,23 +8,29 @@ ENV DEBIAN_FRONTEND=noninteractive \
     TRITON_DISABLE="1" \
     BNB_DISABLE_TRITON="1" \
     USE_TORCH="1" \
-    BITSANDBYTES_NOWELCOME="1"
-
-ENV HF_TOKEN=""
+    BITSANDBYTES_NOWELCOME="1" \
+    HF_TOKEN=""
 
+# Install system dependencies including Python and pip
 RUN apt-get update && apt-get install -y \
-    git wget curl \
+    git wget curl python3 python3-pip python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Make sure pip is properly linked
+RUN ln -sf /usr/bin/python3 /usr/bin/python && \
+    ln -sf /usr/bin/pip3 /usr/bin/pip
+
+# Set working directory
 WORKDIR /app
 
+# Create cache directories with proper permissions
 RUN mkdir -p /tmp/hf_cache && chmod -R 777 /tmp/hf_cache
 
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
-RUN pip install --no-cache-dir transformers accelerate fastapi uvicorn huggingface_hub protobuf
-RUN pip install --no-cache-dir scipy
-RUN pip install --no-cache-dir bitsandbytes
+# Install packages directly
+RUN python -m pip install --no-cache-dir --upgrade pip setuptools wheel
+RUN python -m pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+RUN python -m pip install --no-cache-dir transformers accelerate fastapi uvicorn huggingface_hub protobuf
+RUN python -m pip install --no-cache-dir scipy bitsandbytes
 
 # Expose API port
 EXPOSE 7860