FROM python:3.10-slim

RUN apt-get update && apt-get install -y \
    git cmake build-essential \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app

COPY . .

RUN pip install --no-cache-dir \
    torch \
    torchvision \
    numpy==1.23.5 \
    "Pillow>=9.0.0" \
    scikit-learn==1.1.3 \
    lightgbm==3.3.5 \
    pandas==1.5.3 \
    "huggingface-hub==0.19.4" \
    "gradio==4.7.1" \
    "jinja2==3.1.2" \
    "starlette==0.27.0" \
    "uvicorn==0.22.0" \
    lief==0.12.3 \
    git+https://github.com/elastic/ember.git

CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]