FROM python:3.11-slim

WORKDIR /app

COPY requirements_hf.txt .

RUN apt-get update && apt-get install -y libgomp1 && rm -rf /var/lib/apt/lists/*
RUN pip install --upgrade pip && \
    pip install --no-cache-dir -r requirements_hf.txt


COPY streamlit_app.py .
COPY models/mobilenetv3_catsdogs_quantized_static.onnx ./models/
COPY .streamlit ./.streamlit


RUN mkdir -p /app/.streamlit && chmod -R 777 /app/.streamlit


ENV STREAMLIT_CONFIG_DIR=/app/.streamlit
ENV STREAMLIT_CACHE_DIR=/app/.streamlit/cache
ENV HOME=/app

EXPOSE 7860

CMD ["streamlit", "run", "streamlit_app.py", "--server.port=7860", "--server.address=0.0.0.0"]