Update Dockerfile

Dockerfile

@@ -1,54 +1,27 @@
-# HUGGINGFACE SPACES - DOCKER
 FROM python:3.11-slim
 
-# Argumento: URL directa de descarga (usar "resolve", no "blob")
-ARG MODEL_URL="https://cas-bridge.xethub.hf.co/xet-bridge-us/67fddfc6647578a554f63c8a/8fc299b2c197558dd063213df3475356adda52923314428388d7f7f66fcdd5cc?X-Amz-Algorithm=AWS4-HMAC-SHA256&X-Amz-Content-Sha256=UNSIGNED-PAYLOAD&X-Amz-Credential=cas%2F20250811%2Fus-east-1%2Fs3%2Faws4_request&X-Amz-Date=20250811T001642Z&X-Amz-Expires=3600&X-Amz-Signature=3b06606d4e5a4ec2f8e00eb18c4b67f39b2015deb4f8b5d19b122e2f6f431cba&X-Amz-SignedHeaders=host&X-Xet-Cas-Uid=public&response-content-disposition=inline%3B+filename*%3DUTF-8%27%27ggml-model-i2_s.gguf%3B+filename%3D%22ggml-model-i2_s.gguf%22%3B&x-id=GetObject&Expires=1754875002&Policy=eyJTdGF0ZW1lbnQiOlt7IkNvbmRpdGlvbiI6eyJEYXRlTGVzc1RoYW4iOnsiQVdTOkVwb2NoVGltZSI6MTc1NDg3NTAwMn19LCJSZXNvdXJjZSI6Imh0dHBzOi8vY2FzLWJyaWRnZS54ZXRodWIuaGYuY28veGV0LWJyaWRnZS11cy82N2ZkZGZjNjY0NzU3OGE1NTRmNjNjOGEvOGZjMjk5YjJjMTk3NTU4ZGQwNjMyMTNkZjM0NzUzNTZhZGRhNTI5MjMzMTQ0MjgzODhkN2Y3ZjY2ZmNkZDVjYyoifV19&Signature=PayZUFOSXeiwEGaTW%7EIEaTnkx22AxEeCh8x4hpVTTPl2et8bj3GeVJQVNcDyxvL8h2FlXYF7FSpWONnIe65LSxvpXl9t1h2g%7Eq21Ye%7Eb78Ce3UXU0xXKovLCDIKg8vApJuVM6zDkNnZcjo8N%7Ekwaq2btWn9av1IWxvYQvmLvPBAkg3PRxXDpz0xUPAue2UAf0BXfyIMniCqgHsUSpalHQohMVkZlaBzziCdDTS42C%7EcDhipKwriQ-oXDCgmfpcSir6gJ25eI2prO4fwzY7V6FAT3QJ9dFWBwd%7EycyCkXDU487s%7Es79Bacl5Q0KVfGLjyvhkUZ9%7Eg8LZGCkgiYVNLAQ__&Key-Pair-Id=K2L8F4GPSG1IFC"
-
-# Paquetes necesarios: aria2 (descarga paralela), xz-utils (compresión), coreutils (env vars/du), ca-certificates
+# Instala dependencias: huggingface_hub para descarga, xz-utils para compresión
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-      aria2 xz-utils ca-certificates && \
+        xz-utils && \
+    pip install --no-cache-dir huggingface_hub && \
     rm -rf /var/lib/apt/lists/*
 
-# Directorio de trabajo y datos
-WORKDIR /workspace
+# Directorios
 RUN mkdir -p /data
+WORKDIR /data
 
-# Script de arranque: descarga, comprime, sirve el server
-# - Usa aria2c con 16 conexiones y segmento 1MiB
-# - Comprime con xz al máximo (-9e) y todos los hilos (-T0) a través de tar -I
-# - Publica el archivo por HTTP en /data
-RUN printf '%s\n' \
-'#!/usr/bin/env bash' \
-'set -euo pipefail' \
-'MODEL_URL="${MODEL_URL:-}"' \
-'OUT_DIR="/data"' \
-'RAW="${OUT_DIR}/ggml-model-i2_s.gguf"' \
-'TXZ="${OUT_DIR}/ggml-model-i2_s.gguf.tar.xz"' \
-'' \
-'echo "[1/3] Descargando modelo..."' \
-'if [ ! -s "${RAW}" ]; then' \
-'  aria2c -x16 -s16 -k1M -o "$(basename "${RAW}")" -d "${OUT_DIR}" "${MODEL_URL}"' \
-'else' \
-'  echo "Saltando descarga: existe ${RAW}"' \
-'fi' \
-'' \
-'echo "[2/3] Comprimiendo a tar.xz (máxima compresión, multihilo)..."' \
-'if [ ! -s "${TXZ}" ]; then' \
-'  # Nota: -I permite pasar el pipeline completo de xz; -9e = máximo, -T0 = todos los hilos' \
-'  tar -C "${OUT_DIR}" -c -f "${TXZ}" -I "xz -9e -T0" "$(basename "${RAW}")"' \
-'else' \
-'  echo "Saltando compresión: existe ${TXZ}"' \
-'fi' \
-'' \
-'echo "[3/3] Tamaños:"' \
-'du -h "${RAW}" || true' \
-'du -h "${TXZ}" || true' \
-'' \
-'echo "Listo. Sirviendo /data en http://0.0.0.0:7860/"' \
-'cd "${OUT_DIR}"' \
-'python -m http.server 7860' \
-> /usr/local/bin/run.sh && chmod +x /usr/local/bin/run.sh
+# Script de arranque
+RUN echo '#!/usr/bin/env python3' > /run.py && \
+    echo 'import os' >> /run.py && \
+    echo 'from huggingface_hub import hf_hub_download' >> /run.py && \
+    echo 'print("[1/3] Descargando .gguf...")' >> /run.py && \
+    echo 'raw_path = hf_hub_download("microsoft/bitnet-b1.58-2B-4T-gguf", "ggml-model-i2_s.gguf", local_dir="/data")' >> /run.py && \
+    echo 'print("[2/3] Comprimiendo con xz (máxima compresión)...")' >> /run.py && \
+    echo 'os.system("tar -C /data -c -f /data/model.tar.xz -I \\"xz -9e -T0\\" ggml-model-i2_s.gguf")' >> /run.py && \
+    echo 'print("[3/3] Servidor HTTP en puerto 7860:")' >> /run.py && \
+    echo 'os.chdir("/data")' >> /run.py && \
+    echo 'os.system("python3 -m http.server 7860")' >> /run.py
 
 EXPOSE 7860
-CMD ["/usr/local/bin/run.sh"]
+CMD ["python3", "/run.py"]