Dockerfile

FROM python:3.10-slim

WORKDIR /app

COPY . .

RUN pip install --upgrade pip && \
    pip install flask flask-cors numpy Pillow matplotlib scikit-learn scipy && \
    pip install torch==2.2.2 torchvision==0.17.2 && \
    pip install "transformers==4.40.2" "numpy<2" accelerate

EXPOSE 7860

CMD ["python", "app.py"]