Dockerfile

# Use the official Python 3.11 image
FROM python:3.11

# Copy the current directory contents into the container at .
COPY . .

# Set the working directory to /
WORKDIR /
ENV HF_HOME=/app/huggingface_cache 
# Set environment variables
ENV HUGGINGFACE_TOKEN=""
RUN mkdir -p /app/huggingface_cache && chmod -R 777 /app/huggingface_cache
RUN mkdir -p /output_images && chmod -R 777 /output_images


# Install requirements.txt 
RUN pip install --no-cache-dir --upgrade -r /requirements.txt

# Start the FastAPI app on port 7860, the default port expected by Spaces
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]