Update Dockerfile

Dockerfile

@@ -1,20 +1,21 @@
-FROM huggingface/transformers-pytorch-gpu:latest
+FROM python:3.10-slim
 
 WORKDIR /app
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y \
-    python3-dev \
     build-essential \
     git \
+    python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements first to leverage Docker cache
-COPY requirements.txt .
+# Upgrade pip and install basic requirements
+RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir torch transformers accelerate safetensors
 
-# Upgrade pip and install requirements
-RUN pip install --no-cache-dir -U pip && \
-    pip install --no-cache-dir -r requirements.txt
+# Copy and install requirements
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy the rest of the application
 COPY . .