#!/bin/bash

# Setup script for Hugging Face Space
echo "=== DittoTalkingHead Setup Script ==="

# Create necessary directories
mkdir -p checkpoints/ditto_cfg
mkdir -p tmp
mkdir -p output

# Install system dependencies if needed
# apt-get update && apt-get install -y ffmpeg

# Run model download (PyTorch models)
echo "Starting model download (PyTorch models)..."
python -c "from model_manager import ModelManager; manager = ModelManager(use_pytorch=True); manager.setup_models()"

echo "Setup complete!"