Dockerfile

FROM python:3.10-slim

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

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Clone correct TinyLLaVA source
RUN git clone --depth 1 https://github.com/DLCV-BUAA/TinyLLaVABench.git /tmp/tlv && \
    cp -r /tmp/tlv/tinyllava ./tinyllava && \
    rm -rf /tmp/tlv

# Check existence (optional debug)
RUN test -d ./tinyllava

COPY app.py .

EXPOSE 8501

ENV STREAMLIT_SERVER_HEADLESS=true \
    STREAMLIT_SERVER_PORT=8501 \
    STREAMLIT_SERVER_ADDRESS=0.0.0.0

CMD ["streamlit", "run", "app.py"]