download_models.py

import os
import time
import logging

logger = logging.getLogger(__name__)

def download_models():
    """Download required models for the application"""
    start_time = time.time()
    logger.info("Starting model download process")
    
    try:
        # Create models directory if it doesn't exist
        os.makedirs("models", exist_ok=True)
        
        logger.info(f"Model setup completed in {time.time() - start_time:.2f} seconds")
    except Exception as e:
        logger.error(f"Error in model setup: {str(e)}")
        raise

if __name__ == "__main__":
    download_models()