docs: add prefetch model description

README.md

@@ -12,7 +12,7 @@ All relevant links can be found here:
 
 `pip install --upgrade ssbdplus`  
 
-###  Models and their description: [MODELS.md](./docs/MODELS.md) 
+###  Models and their descriptions: [MODELS.md](./docs/MODELS.md) 
 
 ```py
 from ssbdplus.pipeline import SSBDPipeline

docs/MODELS.md

@@ -1,15 +1,20 @@
 ## SSBDPLUS - Models
 
+### M1 - Preprocessing: Prefetch
+![Prefetch image](./images/SSBDPreFetchArch.png)
+
+Code and documentation for this model can be found [here](../src/ssbdplus/prefetch.py)
+
 ### M1 - Action presence detection
 ![M1 image](./images/SSBDM1.png)  
 
-Code and documentation for this model can be found [here](../src/m1.py)
+Code and documentation for this model can be found [here](../src/ssbdplus/m1.py)
 
 ### M2 - Action type classification
 ![M2 image](./images/SSBDM2.png)
 
-Code and documentation for this model can be found [here](../src/m2.py)
+Code and documentation for this model can be found [here](../src/ssbdplus/m2.py)
 
 ## Running the pipeline
 
-To use the model used in the paper for your task, you can initialise an object defined in the [pipeline file](../src/pipeline.py)
+To use the model used in the paper for your task, you can initialise an object defined in the [pipeline file](../src/ssbdplus/pipeline.py)

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = ssbdplus
-version = 2.1.0
+version = 2.2.0
 author = SARL, IIITB
 author_email = [email protected]
 description = Models pertaining to the paper, Introducing SSBD+ Dataset with a Convolutional Pipeline for detecting Self-Stimulatory Behaviours in Children using raw videos