Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-# Dockerfile (VERSÃO FINAL COM SKIP CUDA BUILD)
+# Dockerfile (VERSÃO FINAL PARA API FASTAPI)
 
 # 1. COMEÇAR COM A BASE CORRETA
 FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
@@ -32,15 +32,14 @@ RUN conda update -n base -c defaults conda
 # 8. CRIAR O AMBIENTE CONDA
 RUN conda env create -f environment.yml && conda clean --all -y
 
-# 9. !!! A MÁGICA FINAL !!! INSTALAR FLASH_ATTN SEM COMPILAR CUDA
+# 9. INSTALAR FLASH_ATTN SEM COMPILAR CUDA (PARA EVITAR ERRO DE MEMÓRIA)
 RUN conda run -n seedvr env FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE \
     pip install "flash_attn==2.5.9.post1" --no-build-isolation --no-cache-dir
 
 # 10. BAIXAR O MODELO
-# (O script de inferência espera que estejamos no diretório /app/SeedVR)
 WORKDIR /app/SeedVR
 RUN huggingface-cli download ByteDance-Seed/SeedVR2-3B --local-dir ckpts --local-dir-use-symlinks False
 
-# 11. DEFINIR UM COMANDO DE "ESPERA"
-# Este comando simplesmente mantém o contêiner rodando para que possamos usar o terminal.
-CMD ["tail", "-f", "/dev/null"]
+# 11. DEFINIR O COMANDO FINAL PARA INICIAR A API
+WORKDIR /app
+CMD ["conda", "run", "-n", "seedvr", "uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]