Update README.md

README.md

@@ -13,7 +13,7 @@ pipeline_tag: token-classification
 # 🦠 MicrobELP — Microbiome Entity Recognition
 
 MicrobELP is a deep learning model for Microbiome Entity Recognition, identifying microbial entities (bacteria, archaea, fungi) in biomedical and scientific text.
-It is part of the [microbELP](https://github.com/omicsNLP/microbELP) toolkit and has been optimised for GPU inference.
+It is part of the [microbELP](https://github.com/omicsNLP/microbELP) toolkit and has been optimised for CPU and GPU inference.
 
 This model enables automated extraction of microbiome names from unstructured text, facilitating microbiome-related text mining and literature curation.
 
@@ -68,7 +68,7 @@ pip install ./microbELP
 
 It is recommended to install in an isolated environment due to dependencies.
 
-Example Usage (GPU model)
+Example Usage
 
 ```python
 from microbELP import microbiome_DL_ner
@@ -103,6 +103,8 @@ Output:
 ```
 Each element in the output corresponds to one input text, containing recognised microbiome entities and their text locations.
 
+There is one optional parameter to this function called `cpu` <type 'bool'>, the default value is False, i.e. runs on a GPU if any are available. If you want to force the usage of the CPU, you will need to use `microbiome_DL_ner(input_list, cpu = True)`.
+
 ---
 
 ## 📘 Model Details