Update Dockerfile

Dockerfile

@@ -1,34 +1,28 @@
-# Use official Python slim image
 FROM python:3.10-slim
 
-# Install system deps (git for clone)
 RUN apt-get update && \
     apt-get install -y git && \
     rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
 
-# Install Python deps first
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# ✅ Clone the real LLaVA repo for tinyllava
-RUN git clone --depth 1 https://github.com/haotian-liu/LLaVA.git /tmp/llava && \
-    cp -r /tmp/llava/tinyllava ./tinyllava && \
-    rm -rf /tmp/llava
+# 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
 
-# Confirm it's there (debug)
-RUN ls -l tinyllava
+# Check existence (optional debug)
+RUN test -d ./tinyllava
 
-# Add your app
 COPY app.py .
 
-# Expose Streamlit port
 EXPOSE 8501
 
-# Streamlit config for Docker
-ENV STREAMLIT_SERVER_HEADLESS=true
-ENV STREAMLIT_SERVER_PORT=8501
-ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
+ENV STREAMLIT_SERVER_HEADLESS=true \
+    STREAMLIT_SERVER_PORT=8501 \
+    STREAMLIT_SERVER_ADDRESS=0.0.0.0
 
 CMD ["streamlit", "run", "app.py"]