Add model summaries to Python code