Revert to python:3.10-slim (was working) + health server prevents timeout

Dockerfile

@@ -1,9 +1,7 @@
-FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
+FROM python:3.10-slim
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y python3 python3-pip && rm -rf /var/lib/apt/lists/*
-
 ENV HF_HOME=/tmp/hf_cache
 ENV TRANSFORMERS_CACHE=/tmp/hf_cache
 ENV TORCH_HOME=/tmp/torch_cache
@@ -16,10 +14,8 @@ ENV PYTHONUNBUFFERED=1
 RUN mkdir -p /tmp/hf_cache /tmp/torch_cache /tmp/cache /tmp/home && \
     chmod -R 777 /tmp/hf_cache /tmp/torch_cache /tmp/cache /tmp/home
 
-# Install PyTorch with CUDA 12.1 support explicitly
-RUN pip3 install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cu121
-
-RUN pip3 install --no-cache-dir \
+RUN pip install --no-cache-dir \
+    torch \
     transformers \
     trl \
     peft \