- Update readme

README.md

@@ -49,6 +49,15 @@ from transformers import AutoModel
 humit_tagger = AutoModel.from_pretrained("Humit-Oslo/humit-tagger-small", trust_remote_code=True)
 ```
 
+While creating the model, batch\_size and device can be given as parameters:
+```python
+humit_tagger = AutoModel.from_pretrained("Humit-Oslo/humit-tagger-xs", trust_remote_code=True, batch_size=16, device="cuda")
+```
+
+Here the batch size can be a power of 2, and can be set to higher values, such as 32 or 64, if the model will be loaded on a powerful GPU.
+If the device is not set then the model will try to locate itself on the first CUDA device if it exists, otherwise on CPU.
+A specific device can be set such as "cuda:0" or "cuda:1".
+
 ## Functions and parameters
 
 The model provides two functions: tag and identify\_language.