Tentativa com nova bib de segmentação.

Dockerfile

@@ -16,21 +16,11 @@ COPY requirements.txt .
 RUN pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
-#PRÉ-BAIXA OS MODELOS (CONVNEXT-LARGE)
-RUN python -c "from huggingface_hub import snapshot_download; print('Downloading ConvNext models (snapshot)...'); snapshot_download(repo_id='facebook/convnext-large-224-22k-1k'); print('ConvNext models downloaded.')"
-#CORREÇÃO: Força o download do backgroundremover com imagem dummy
-#(Usando sintaxe de múltiplas linhas para o 'try...except')
-RUN python -c """
-import backgroundremover
-from PIL import Image
-print('Downloading backgroundremover model (u2net)...')
-img = Image.new('RGB', (1, 1), 'black')
-try:
-    backgroundremover.remove(img)
-except Exception:
-    pass
-print('Model download triggered.')
-"""
+#PRÉ-BAIXA OS MODELOS
+#Copia o script de download para a imagem
+COPY download_models.py .
+#Executa o script de download
+RUN python download_models.py
 
 #Copia o código da aplicação
 COPY . .

download_models.py

@@ -0,0 +1,24 @@
+import os
+import sys
+from huggingface_hub import snapshot_download
+import backgroundremover
+from PIL import Image
+
+#1. Baixa o modelo ConvNext
+print("Downloading ConvNext-LARGE model (snapshot)...")
+snapshot_download(repo_id='facebook/convnext-large-22k-1k')
+print("ConvNext-LARGE model downloaded.")
+
+#2. Baixa o modelo backgroundremover
+print("Downloading backgroundremover model (u2net)...")
+#Usa o truque da imagem falsa de 1x1 para forçar o download
+img = Image.new('RGB', (1, 1), 'black')
+try:
+    backgroundremover.remove(img)
+except Exception as e:
+    # A execução vai falhar na imagem 1x1, o que é esperado.
+    # Nós não nos importamos, contanto que o download tenha sido disparado.
+    print(f"Disparado o download (ignora erro esperado: {e})")
+    pass
+    
+print("Model downloads finished.")