Update README.md

README.md

@@ -26,3 +26,14 @@ Use the `Color-Invariant-Skin-Segmentation` module to generate masks, following
 ### In-paint the images
 
 Use the `BrushNet` module to in-paint the images from FACET and/or CC3M (see the respective scripts in `BrushNet/examples/brushnet/inapaint_*.py`.
+
+
+## Zero-shot classification
+
+Use the `zero_shot_classification.py` script to test the occupation classification accuracy of different CLIP models (for different in-painting setups, i.e. PP, PP*, WB, etc.). To run it, first install PyTorch and the following dependencies:
+`pip install open_clip`
+`pip install git+https://github.com/openai/CLIP.git`
+`pip install tqdm`
+`pip install numpy`
+`pip install pandas`
+`pip install pillow`