Improve Docker build process and add startup script

- Add git-lfs to system dependencies for model downloading
- Create start.sh script to handle model downloads at runtime
- Update Dockerfile to use startup script instead of direct python execution
- This should resolve build timeouts and missing model issues

Dockerfile

@@ -24,6 +24,7 @@ RUN apt-get update && apt-get install -y \
     libtcmalloc-minimal4 \
     ffmpeg \
     tzdata \
+    git-lfs \
     && ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
@@ -36,6 +37,7 @@ ENV PATH="/home/user/.local/bin:$PATH"
 ENV PYTHONPATH=/app
 ENV GRADIO_SERVER_NAME=0.0.0.0
 ENV GRADIO_SERVER_PORT=7860
+ENV HF_HOME=/app/cache
 
 # Set working directory
 WORKDIR /app
@@ -48,10 +50,13 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 COPY --chown=user . /app
 
 # Create necessary directories
-RUN mkdir -p pretrained_models outputs
+RUN mkdir -p pretrained_models outputs cache
+
+# Make scripts executable
+RUN chmod +x download_models.sh start.sh
 
 # Expose port (required by HF Spaces to be 7860)
 EXPOSE 7860
 
-# Start the application
-CMD ["python", "app.py"]
+# Start the application using startup script
+CMD ["./start.sh"]

start.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+
+echo "Starting AI Avatar Chat application..."
+
+# Check if models exist, if not download them
+if [ ! -d "pretrained_models/OmniAvatar-14B" ]; then
+    echo "Models not found, downloading..."
+    ./download_models.sh
+else
+    echo "Models already exist, skipping download..."
+fi
+
+echo "Starting Python application..."
+python app.py