Update Dockerfile

Dockerfile

@@ -2,37 +2,51 @@ FROM python:3.11-slim
 
 ENV PIP_NO_CACHE_DIR=1 \
     PYTHONDONTWRITEBYTECODE=1 \
-    PYTHONUNBUFFERED=1
-
-ENV APP_CACHE=/tmp/appcache \
+    PYTHONUNBUFFERED=1 \
+    APP_CACHE=/tmp/appcache \
     XDG_CACHE_HOME=/tmp/appcache \
     HF_HOME=/tmp/appcache/hf \
     HUGGINGFACE_HUB_CACHE=/tmp/appcache/hf \
     TORCH_HOME=/tmp/appcache/torch \
     OPENCLIP_CACHE_DIR=/tmp/appcache/open_clip \
-    HF_HUB_ENABLE_HF_TRANSFER=0
+    HF_HUB_ENABLE_HF_TRANSFER=0 \
+    OMP_NUM_THREADS=2 \
+    MKL_NUM_THREADS=2 \
+    NUM_THREADS=2
 
+# deps del sistema: OpenMP y build tools
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential git && \
+    build-essential git libgomp1 && \
     rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# deps de proyecto
+# deps de proyecto (ligeras)
 COPY requirements.txt ./
-RUN pip install --upgrade pip && pip install -r requirements.txt
+RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
+
+# PyTorch CPU wheels oficiales
+RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu \
+    torch torchvision
 
-# PyTorch CPU y OpenCLIP
-RUN pip install --index-url https://download.pytorch.org/whl/cpu torch torchvision
-RUN pip install open_clip_torch
+# (opcional) asegurar que open_clip este instalado (si no viene por reqs)
+# RUN pip install --no-cache-dir open_clip_torch
 
 # copiar codigo y embeddings
-COPY server1.py ./
-COPY text_embeddings_modelos_bigg.pt ./
-COPY text_embeddings_bigg.pt ./
+COPY server1.py ./server1.py
+COPY text_embeddings_modelos_bigg.pt ./text_embeddings_modelos_bigg.pt
+COPY text_embeddings_bigg.pt ./text_embeddings_bigg.pt
+
+# (opcional) precarga ligera para evitar cold-start de dependencias
+# No hace forward pesado ni cambia precision
+# RUN python - <<'PY'\n\
+# import open_clip\n\
+# open_clip.create_model_and_transforms('hf-hub:timm/PE-Core-bigG-14-448', pretrained=None)\n\
+# print('open_clip model meta loaded')\n\
+# PY
 
-# Puerto de Spaces
 ENV PORT=7860
 EXPOSE 7860
 
-CMD ["uvicorn", "server1:app", "--host", "0.0.0.0", "--port", "7860"]
+# uvloop + httptools (vienen con uvicorn[standard]); 1 worker evita duplicar el modelo
+CMD ["uvicorn", "server1:app", "--host", "0.0.0.0", "--port", "7860", "--loop", "uvloop", "--http", "httptools", "--workers", "1", "--no-server-header", "--timeout-keep-alive", "5"]