media-unmasked-api / Dockerfile
wozwize's picture
adding test files. updating exception messages
212d694
raw
history blame
620 Bytes
# Use Python Slim Image
FROM python:3.10-slim
# Set working directory inside container
WORKDIR /app
# Create a writable cache directory for Hugging Face
RUN mkdir -p /app/.cache/huggingface/hub && chmod -R 777 /app/.cache
# Copy dependencies
COPY requirements.txt .
RUN pip install -r requirements.txt
# Copy all files
COPY . .
# Set PYTHONPATH (simplified)
ENV PYTHONPATH=/app
# Set the Hugging Face transformers cache directory
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
# Expose FastAPI's default port
EXPOSE 7860
# Start FastAPI
CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]