Dockerfile

# Use your existing image from Docker Hub
FROM shahrukhsiddiq/diffusion-api:latest

# Set working directory
WORKDIR /app

# Expose the port that your FastAPI runs on
EXPOSE 8000

# Optional environment vars (improves caching on HF)
ENV TRANSFORMERS_CACHE=/data
ENV HF_HOME=/data

# Ensure correct entrypoint for FastAPI
CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "8000"]