Update README.md

README.md

@@ -46,11 +46,15 @@ source venv/bin/activate
 pip install -e .
 ```
 
+### Infernece
+
+We provide an easy-to-use inference script that automatically downloads the weights from Hugging Face. You just need to provide a path to folder with Sentinel-2 L2A files (all 12 bands) and `class_names` or a `query`.
+
 Run zero-shot classification with:
 ```shell
 python inference.py --run-classification \
     --model-name "Llama3-MS-CLIP-Base" \
-    --images "path/to/sentinel2_files.tif" \
+    --images "path/to/sentinel2_l2a_files/" \
     --class-names "class 1" "class 2" "class 3"
 ```
 
@@ -58,7 +62,7 @@ Run text-to-image retrieval with:
 ```shell
 python inference.py --run_retrieval\
     --model-name "Llama3-MS-CLIP-Base" \
-    --images "path/to/sentinel2_files.tif" \
+    --images "path/to/sentinel2_l2a_files/" \
     --query "Your query text" \
     --top-k 5  # Number of retrieved images
 ```