Optimize build: remove heavy deps (torch_scatter/transformers/biopython) — ETA <5min [2026-03-27 14:29 UTC]

Dockerfile

@@ -1,8 +1,8 @@
 FROM python:3.11-slim
 
-# System deps (added curl for HEALTHCHECK)
+# System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential git curl && \
+    curl && \
     rm -rf /var/lib/apt/lists/*
 
 # Add a non-root user (Hugging Face Spaces runs as 1000)
@@ -10,13 +10,10 @@ RUN useradd -m -u 1000 appuser
 
 WORKDIR /app
 
-# Step 1: Install torch CPU
-RUN pip install --no-cache-dir torch==2.6.0
+# Install torch CPU-only (slim, no CUDA)
+RUN pip install --no-cache-dir torch==2.6.0+cpu --index-url https://download.pytorch.org/whl/cpu
 
-# Step 2: Install torch extensions from PyG wheels
-RUN pip install --no-cache-dir torch_scatter torch_sparse -f https://data.pyg.org/whl/torch-2.6.0+cpu.html
-
-# Step 3: Other Python deps
+# Install other Python deps
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
@@ -27,12 +24,11 @@ COPY --chown=appuser:appuser . .
 ENV PYTHONPATH=/app:$PYTHONPATH \
     HOME=/home/appuser
 
-# Streamlit config (placed in the correct HOME directory)
+# Streamlit config
 RUN mkdir -p /home/appuser/.streamlit && \
     printf '[server]\nheadless = true\nport = 7860\nenableCORS = false\nenableXsrfProtection = false\n' > /home/appuser/.streamlit/config.toml && \
     chown -R appuser:appuser /home/appuser/.streamlit /app
 
-# Switch to the non-root user
 USER appuser
 
 EXPOSE 7860

requirements.txt

@@ -1,17 +1,11 @@
 # PETIMOT Explorer — HuggingFace Spaces
-# Note: torch, torch_scatter, torch_sparse installed separately in Dockerfile
 streamlit>=1.30.0
 stmol>=0.0.9
 py3Dmol>=2.0.0
 plotly>=5.18.0
 pandas>=2.0.0
 numpy>=1.24.0
-torch_geometric>=2.0.0
-transformers>=4.30.0
-sentencepiece>=0.1.99
 scipy>=1.10.0
-biopython>=1.80
 requests>=2.28.0
 tqdm>=4.65.0
-typer>=0.9.0
 huggingface_hub>=0.20.0