Update Dockerfile

Dockerfile

@@ -1,36 +1,27 @@
-# Use official Python image
 FROM python:3.10-slim
 
-# Install git to clone the repo
 RUN apt-get update && \
     apt-get install -y git && \
-    rm -rf /var/lib/apt/lists/* && \
-    apt-get clean
+    rm -rf /var/lib/apt/lists/*
 
-# Set working dir
 WORKDIR /app
 
-# Copy only requirements first
+# Copy requirements and install dependencies
 COPY requirements.txt .
-
-# Install Python packages
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Clone the LLaVA repo and copy only tinyllava folder
-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 TinyLLaVA Factory and copy tinyllava module
+RUN git clone --depth 1 https://github.com/TinyLLaVA/TinyLLaVA_Factory.git /tmp/tlv && \
+    cp -r /tmp/tlv/tinyllava ./tinyllava && \
+    rm -rf /tmp/tlv
 
-# Copy the rest of your code
+# Copy your app
 COPY . .
 
-# Expose Streamlit port
 EXPOSE 7860
 
-# Streamlit config for Docker
 ENV STREAMLIT_SERVER_HEADLESS=true
 ENV STREAMLIT_SERVER_PORT=7860
 ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
 
-# Run
 CMD ["streamlit", "run", "app.py"]