FROM python:3.10-slim

# Install g++
RUN apt-get update && apt-get install -y g++

WORKDIR /app

COPY ./ /app

# For GPU support:
RUN pip3 install --extra-index-url https://download.pytorch.org/whl/cu118 torch torchvision 

RUN pip install -r requirements.txt


CMD fastapi run --reload --host=0.0.0.0 --port=7860