Dockerfile

FROM python:3.10-slim

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

WORKDIR /app

COPY . .

RUN pip install \
    flask \
    requests \
    pandas \
    scikit-learn \
    torch \
    transformers \
    peft

RUN chmod -R 777 /app

RUN mkdir -p /app/cache && chmod -R 777 /app /app/cache

ENV TRANSFORMERS_CACHE=/app/cache

EXPOSE 7860

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