docs: update usage example

README.md

@@ -45,21 +45,45 @@ pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp
 
 ## Examples
 
-### Load pre-trained model
+### How to load pre-trained model for Masked Language Modeling
 ```python
-from transformers import AutoTokenizer, BigBirdForMaskedLM
+from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse')
-model = BigBirdForMaskedLM.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse')
+model = AutoModel.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse', trust_remote_code=True)
+
 ```
 
+### How to load pre-trained model to fine-tune it on classification task
+Get model class from GENA-LM repository:
+```bash
+git clone https://github.com/AIRI-Institute/GENA_LM.git
+```
 
-### How to load the model to fine-tune it on classification task
 ```python
-from transformers import AutoTokenizer, BigBirdForSequenceClassification
+from GENA_LM.src.gena_lm.modeling_bert import BertForSequenceClassification
+from transformers import AutoTokenizer
 
 tokenizer = AutoTokenizer.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse')
-model = BigBirdForSequenceClassification.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse')
+model = BertForSequenceClassification.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse')
+```
+or you can just download [modeling_bert.py](https://github.com/AIRI-Institute/GENA_LM/tree/main/src/gena_lm) and put it close to your code.
+
+OR you can get model class from HuggingFace AutoModel:
+```python
+from transformers import AutoTokenizer, AutoModel
+model = AutoModel.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse', trust_remote_code=True)
+gena_module_name = model.__class__.__module__
+print(gena_module_name)
+import importlib
+# available class names:
+# - BertModel, BertForPreTraining, BertForMaskedLM, BertForNextSentencePrediction,
+# - BertForSequenceClassification, BertForMultipleChoice, BertForTokenClassification,
+# - BertForQuestionAnswering
+# check https://huggingface.co/docs/transformers/model_doc/bert
+cls = getattr(importlib.import_module(gena_module_name), 'BertForSequenceClassification')
+print(cls)
+model = cls.from_pretrained('AIRI-Institute/gena-lm-bigbird-base-sparse', num_labels=2)
 ```
 
 ## Model description