Update Dockerfile

Dockerfile

@@ -2,7 +2,7 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
-# Install system dependencies
+# Install system dependencies for torch, matplotlib, and transformers
 RUN apt-get update && apt-get install -y \
     libgomp1 \
     gcc \
@@ -11,7 +11,7 @@ RUN apt-get update && apt-get install -y \
 
 # Install Python dependencies
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt && pip cache purge
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application files
 COPY . .