Dockerfile

FROM python:3.11-slim

WORKDIR /app

# Установка системных зависимостей, если нужно
RUN apt-get update && apt-get install -y git gcc && rm -rf /var/lib/apt/lists/*

# Сначала копируем requirements.txt, чтобы использовать кэш
COPY requirements.txt .

# Устанавливаем зависимости
RUN pip3 install --no-cache-dir -r requirements.txt

# === НОВЫЙ БЛОК: Загрузка моделей при сборке ===
# Загрузка модели и токенизатора transformers
RUN python -c "from transformers import AutoTokenizer, AutoModelForQuestionAnswering; \
    AutoTokenizer.from_pretrained('AlexKay/xlm-roberta-large-qa-multilingual-finedtuned-ru'); \
    AutoModelForQuestionAnswering.from_pretrained('AlexKay/xlm-roberta-large-qa-multilingual-finedtuned-ru')"

# Копируем остальные файлы и делаем новую загрузку
COPY . .

EXPOSE 8501

HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

ENTRYPOINT ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"]