Update README.md

README.md

@@ -20,7 +20,7 @@ tags:
 
 ***
 
-# Installation
+## Installation
 
 ```python
 from huggingface_hub import snapshot_download
@@ -33,5 +33,17 @@ local_dir = "./Imagen-South-Park"
 snapshot_download(repo_id, repo_type=repo_type, local_dir=local_dir)
 ```
 
+## Inference
+
+```sh
+!pip install -U imagen-pytorch
+```
+
+```
+import torch
+from imagen_pytorch import Unet, Imagen, ImagenTrainer
+from imagen_pytorch.data import Dataset
+```
+
 ### Project Los Angeles
 ### Tegridy Code 2024