trashnet-server / Dockerfile
froidhj's picture
Update Dockerfile
a242be5 verified
raw
history blame
825 Bytes
# Dockerfile
FROM python:3.11-slim
# Evita cache e força o cache do HF dentro de /app (onde temos permissão)
ENV PIP_NO_CACHE_DIR=1
ENV HF_HOME=/app/hf_cache
# Dependências básicas
RUN apt-get update && apt-get install -y --no-install-recommends \
&& rm -rf /var/lib/apt/lists/*
# Garante que a pasta de cache existe
RUN mkdir -p /app/hf_cache
# Pasta de trabalho
WORKDIR /app
# Instala Torch **CPU** (evita baixar CUDA gigante)
RUN pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cpu torch
# Copia e instala os demais requisitos
COPY requirements.txt /app/
RUN pip install --no-cache-dir -r requirements.txt
# Copia o app
COPY app.py /app/
# Porta usada pelo Spaces
ENV PORT=7860
EXPOSE 7860
# Inicia o servidor FastAPI
CMD ["sh", "-c", "uvicorn app:app --host 0.0.0.0 --port $PORT"]