Add Available classes section to model card

README.md

@@ -57,6 +57,19 @@ Load any ELECTRA variant the same way with `from_weights("kerasformers/<variant>
 | base | [`kerasformers/electra_base_discriminator`](https://huggingface.co/kerasformers/electra_base_discriminator) | [`kerasformers/electra_base_generator`](https://huggingface.co/kerasformers/electra_base_generator) |
 | large | [`kerasformers/electra_large_discriminator`](https://huggingface.co/kerasformers/electra_large_discriminator) | [`kerasformers/electra_large_generator`](https://huggingface.co/kerasformers/electra_large_generator) |
 
+## Available classes
+
+Load any of these from this repo with `from_weights("kerasformers/electra_base_generator")` (or on the fly via the `hf:` prefix). The pretrained backbone is shared; task heads not stored in this checkpoint start randomly initialized, ready for fine-tuning (or load a `hf:` fine-tune).
+
+| Class | Task |
+|---|---|
+| `ElectraMaskedLM` | Masked language modeling (fill-mask) |
+
+```python
+from kerasformers.models.electra import ElectraMaskedLM
+model = ElectraMaskedLM.from_weights("kerasformers/electra_base_generator")
+```
+
 ## Tips
 
 - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.