Update README.md

Author: williamFalcon

README.md

@@ -213,9 +213,12 @@ Lightning has over [40+ advanced features](https://pytorch-lightning.readthedocs
 
 Here are some examples:
 
+<div align="center">
+  <img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/features_2.jpg" max-height="600px">
+</div>
 
 <details>
-  <summary>Train on GPUs without code changes</summary>
+  <summary>Highlighted feature code snippets</summary>
 
   ```python
   # 8 GPUs
@@ -225,27 +228,21 @@ Here are some examples:
   # 256 GPUs
   trainer = Trainer(max_epochs=1, gpus=8, num_nodes=32)
   ```
-</details>
 
-<details>
   <summary>Train on TPUs without code changes</summary>
 
   ```python
   # no code changes needed
   trainer = Trainer(tpu_cores=8)
    ```
-</details>
 
-<details>
   <summary>16-bit precision</summary>
 
   ```python
   # no code changes needed
   trainer = Trainer(precision=16)
    ```
-</details>
 
-<details>
   <summary>Experiment managers</summary>
 
   ```python
@@ -268,37 +265,29 @@ Here are some examples:
 
   # ... and dozens more
    ```
-</details>
 
-<details>
   <summary>EarlyStopping</summary>
 
   ```python
   es = EarlyStopping(monitor='val_loss')
   trainer = Trainer(callbacks=[es])
    ```
-</details>
 
-<details>
   <summary>Checkpointing</summary>
 
   ```python
   checkpointing = ModelCheckpoint(monitor='val_loss')
   trainer = Trainer(callbacks=[checkpointing])
    ```
-</details>
 
-<details>
   <summary>Export to torchscript (JIT) (production use)</summary>
 
   ```python
   # torchscript
   autoencoder = LitAutoEncoder()
   torch.jit.save(autoencoder.to_torchscript(), "model.pt")
    ```
-</details>
 
-<details>
   <summary>Export to ONNX (production use)</summary>
 
   ```python