Fix snippet

README.md

@@ -96,6 +96,18 @@ Now, let's instantiate the model and processor.
 >>> processor = get_processor()
 ```
 
+### Zero-shot classification
+
+```python
+>>> from colipri import ZeroShotImageClassificationPipeline
+>>> pipeline = ZeroShotImageClassificationPipeline(model, processor)
+>>> pipeline(image, ["No lung nodules", "Lung nodules"])
+[
+    {'score': 0.005, 'label': 'No lung nodules'},
+    {'score': 0.995, 'label': 'Lung nodules'}
+]
+```
+
 ### Feature extraction
 
 ```python
@@ -111,23 +123,11 @@ torch.Size([1, 1, 192, 192, 192])
 >>> patch_embeddings.shape
 torch.Size([1, 768, 24, 24, 24])
 >>> with torch.no_grad():
-...     pooled_embeddings = model.encode_image(images_batch, pool=True)
+...     pooled_embeddings = model.encode_image(images_batch, pool=True, project=True)
 >>> pooled_embeddings.shape
 torch.Size([1, 768])
 ```
 
-### Zero-shot classification
-
-```python
->>> from colipri import ZeroShotImageClassificationPipeline
->>> pipeline = ZeroShotImageClassificationPipeline(model, processor)
->>> pipeline(image, ["No lung nodules", "Lung nodules"])
-[
-    {'score': 0.005, 'label': 'No lung nodules'},
-    {'score': 0.995, 'label': 'Lung nodules'}
-]
-```
-
 ## Environmental impact
 
 <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->