Update Dockerfile

Dockerfile

@@ -12,17 +12,25 @@ RUN git clone https://huggingface.co/spaces/Finish-him/prometheus-embedding-gene
 COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-# Define o caminho da cache (ainda uma boa prática)
+# Define o caminho da cache
 ENV HF_HOME=/app/cache/huggingface
 ENV SENTENCE_TRANSFORMERS_HOME=/app/cache/torch
+RUN mkdir -p $HF_HOME && mkdir -p $SENTENCE_TRANSFORMERS_HOME
 
 # Pré-aquece o cache com o modelo e5-large
 RUN python -c "import os; from sentence_transformers import SentenceTransformer; SentenceTransformer('intfloat/multilingual-e5-large', cache_folder=os.environ.get('SENTENCE_TRANSFORMERS_HOME', '/app/cache/torch'))"
 
-# --- ETAPA 3: EXECUTAR O SCRIPT ---
+# --- ETAPA 3: PREPARAR DIRETÓRIOS E COPIAR SCRIPT ---
+
+# *** A CORREÇÃO ESTÁ AQUI ***
+# Cria os diretórios necessários e dá permissão ao usuário padrão (1000)
+RUN mkdir -p /app/dados_extraidos && \
+    mkdir -p /app/output && \
+    chown -R 1000:1000 /app/dados_extraidos /app/output /app/cache
+
 # Copia o script de treinamento
 COPY train.py .
 
+# --- ETAPA 4: EXECUTAR O SCRIPT ---
 # Executa o script e, após o término, mantém o contêiner a correr.
-# Como agora estamos a correr como root, não há necessidade de chown ou de criar pastas manualmente.
 CMD ["sh", "-c", "python train.py && sleep infinity"]