image-denoiser / Dockerfile.api
Kajuto's picture
Initial commit - image denoiser + SR + MLOps stack
8b83582
raw
history blame contribute delete
770 Bytes
# Dockerfile.api — FastAPI inference API
FROM python:3.11-slim
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt \
&& pip install --no-cache-dir \
fastapi \
uvicorn[standard] \
python-multipart \
prometheus-client
COPY models/ models/
COPY training/ training/
COPY app/ app/
COPY api/ api/
COPY data/dataset.py data/dataset.py
COPY config.py .
COPY checkpoints/best_sr_gaussian.pth checkpoints/best_sr_gaussian.pth
COPY checkpoints/best_sr_salt_pepper.pth checkpoints/best_sr_salt_pepper.pth
COPY checkpoints/best_sr_speckle.pth checkpoints/best_sr_speckle.pth
EXPOSE 8000
CMD ["uvicorn", "api.main:app", "--host", "0.0.0.0", "--port", "8000"]