Actualizar enlaces y ajustes para los .onnx

Dockerfile

@@ -1,11 +1,11 @@
-# 1. Usamos 3.11 por estabilidad absoluta con PyTorch y YOLO sin compilar desde cero
+# 1. Usamos 3.11 por estabilidad absoluta con PyTorch, YOLO y ONNX sin compilar desde cero
 FROM python:3.11-slim
 
 # 2. Instalamos dependencias del sistema mínimas. 
-# NOTA: Asumimos que usarás 'opencv-python-headless' en requirements.txt
-# Si estrictamente procesas video, deja ffmpeg. Si son solo imágenes, puedes borrar el RUN completo.
+# libgomp1 es crucial para ONNX Runtime en imágenes slim de Debian
 RUN apt-get update && apt-get install -y --no-install-recommends \
     ffmpeg \
+    libgomp1 \
     && rm -rf /var/lib/apt/lists/*
 
 # --- CONFIGURACIÓN DE HUGGING FACE ---
@@ -13,6 +13,7 @@ RUN useradd -m -u 1000 user
 USER user
 
 # 3. Variables de entorno: PATH y Optimización estricta para CPU
+# ONNX Runtime respetará estas variables para no saturar los hilos
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
     OMP_NUM_THREADS=1 \
@@ -35,5 +36,5 @@ COPY --chown=user . .
 
 EXPOSE 7860
 
-# 7. Arranque
-CMD ["uvicorn", "src.app.app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+# 7. Arranque optimizado para 16GB de RAM (4 workers)
+CMD ["uvicorn", "src.app.app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "4"]

requirements.txt

@@ -23,6 +23,8 @@ matplotlib==3.10.9
 mpmath==1.3.0
 networkx==3.6.1
 numpy==2.4.4
+onnx==1.17.0
+onnxruntime==1.20.0
 opencv-python-headless==4.13.0.92
 packaging==26.2
 pillow==12.2.0
@@ -36,8 +38,8 @@ PySocks==1.7.1
 python-dateutil==2.9.0.post0
 python-multipart==0.0.29
 PyYAML==6.0.3
-s3transfer==0.18.0
 requests==2.34.2
+s3transfer==0.18.0
 scipy==1.17.1
 setuptools==70.2.0
 six==1.17.0

src/app/app.py

@@ -22,15 +22,15 @@ R2_SECRET_KEY = os.getenv("R2_SECRET_KEY")
 model_configs = {
     "VEF": {
         "r2_key": "files/md5/47/4668895cd0778ae1381ce2e3fdfd46",
-        "path": "models/train/VEF_model_13f/weights/best.pt"
+        "path": "models/train/VEF_model_13f/weights/best.onnx"
     },
     "USD": {  
         "r2_key": "files/md5/54/f34a7d848129a2cfbc9f9250b3e9ef",
-        "path": "models/train/USD_model_plus_01/weights/best.pt"
+        "path": "models/train/USD_model_plus_01/weights/best.onnx"
     },
     "INFERENCIA": {
         "r2_key": "files/md5/48/4833ce8e1f80f4c07e4e087feb5823",
-        "path": "models/train/USD_VEF_Model_01j/weights/best.pt"
+        "path": "models/train/USD_VEF_Model_01j/weights/best.onnx"
     }
 }
 
@@ -80,9 +80,9 @@ download_models_from_r2()
 # 4. CARGA DE MODELOS
 # ==========================================
 models = {
-    "USD":        YOLO(model_configs["USD"]["path"]),
-    "VEF":        YOLO(model_configs["VEF"]["path"]),
-    "INFERENCIA": YOLO(model_configs["INFERENCIA"]["path"]),
+    "USD":        YOLO(model_configs["USD"]["path"], task="detect"),
+    "VEF":        YOLO(model_configs["VEF"]["path"], task="detect"),
+    "INFERENCIA": YOLO(model_configs["INFERENCIA"]["path"], task="detect"),
 }
 
 classes = {