docker-v2

Dockerfile

@@ -1,15 +1,31 @@
-# Use uma imagem base oficial com Python
+# Usa uma imagem base oficial com Python
 FROM python:3.9
 
+# Cria um usuário não-root para segurança
+RUN useradd -m -u 1000 user
+USER user
+
+# --- CORREÇÃO INICIA AQUI ---
+
+# Define a pasta de cache dentro do diretório home do usuário.
+# Todas as bibliotecas do Hugging Face usarão este caminho.
+ENV HF_HOME=/home/user/.cache/huggingface
+ENV SENTENCE_TRANSFORMERS_HOME=/home/user/.cache/torch
+
+# Cria a pasta de cache e garante que o usuário tenha permissão para escrever nela.
+RUN mkdir -p $HF_HOME && mkdir -p $SENTENCE_TRANSFORMERS_HOME
+
+# --- CORREÇÃO TERMINA AQUI ---
+
 # Define o diretório de trabalho
 WORKDIR /app
 
 # Copia o arquivo de requerimentos e instala as dependências
-COPY requirements.txt requirements.txt
+COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 # Copia todos os outros arquivos (seu script, seu .zip, etc.)
-COPY . .
+COPY --chown=user . .
 
 # Comando que será executado quando o Space iniciar
 CMD ["python", "train.py"]