Update Dockerfile

Dockerfile

@@ -1,36 +1,33 @@
-FROM python:3.9-slim
+# Use a more specific Python version
+FROM python:3.9.16-slim
 
+# Set working directory and environment variables
 WORKDIR /app
 ENV TRANSFORMERS_CACHE=/app/cache \
-    HF_HOME=/app/cache
+    HF_HOME=/app/cache \
+    PYTHONUNBUFFERED=1
 
-# 1. Install system dependencies (including scipy's required libraries)
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     gcc \
     g++ \
     make \
-    libopenblas-dev \  # Required for scipy
     && rm -rf /var/lib/apt/lists/*
 
-# 2. Create writable cache directory
+# Create writable cache directory
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
-# 3. Install Python packages with COMPATIBLE versions
-RUN pip install --no-cache-dir \
-    scipy \  # Explicitly install first
-    torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 \
-    transformers==4.34.0 \
-    peft==0.5.0 \
-    datasets==2.14.0 \
-    accelerate==0.24.0 \
-    bitsandbytes==0.41.0 \
-    huggingface-hub==0.16.4
+# Install Python packages using a requirements.txt file
+COPY requirements.txt.
+RUN pip install --no-cache-dir -r requirements.txt
 
-# 4. Authenticate with Hugging Face
+# Authenticate with Hugging Face
 ARG HF_TOKEN
 RUN python -c "from huggingface_hub import HfFolder; HfFolder.save_token('$HF_TOKEN')"
 
-COPY train.py .
+# Copy train.py file
+COPY train.py.
 
+# Set command to run train.py
 CMD ["python", "train.py"]