FROM python:3.9-slim

# Install dependencies
RUN pip install fastapi uvicorn tensorflow pillow numpy

# Copy model and app
COPY super_resolution_model.h5 .
COPY app.py .

# Expose port
EXPOSE 8000

# Run the app
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]