NMundhra's picture
Upload folder using huggingface_hub
be7e39c verified
Raw
History Blame Contribute Delete
787 Bytes
# Use the official PyTorch image with CUDA support
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
# Set working directory
WORKDIR /app
# Install system dependencies for image processing (OpenCV, SimpleITK)
RUN apt-get update && apt-get install -y \
libgl1-mesa-glx \
libglib2.0-0 \
&& rm -rf /var/lib/apt/lists/*
# Install core Python libraries
COPY backend/requirements.txt /app/backend/requirements.txt
RUN pip install --no-cache-dir -r backend/requirements.txt
# Copy your codebase into the container
COPY backend /app/backend
# Run the web service on container startup.
# The PORT environment variable is automatically injected by Cloud Run / Heroku / HF Spaces.
ENV PYTHONPATH="/app"
CMD uvicorn backend.oct_analyzer.api:app --host 0.0.0.0 --port ${PORT:-8000}