Commit
·
383a5ca
1
Parent(s):
74ae76c
Update README.md
Browse files
README.md
CHANGED
|
@@ -20,7 +20,7 @@ sys.path.append("PATH_TO_CLONED_PROJECT/MiniALBERT/")
|
|
| 20 |
from minialbert_modeling import MiniAlbertForSequenceClassification, MiniAlbertForTokenClassification
|
| 21 |
```
|
| 22 |
Finally, load the model like a regular model in the transformers library using the below code:
|
| 23 |
-
```
|
| 24 |
# For NER use the below code
|
| 25 |
model = MiniAlbertForTokenClassification.from_pretrained("nlpie/miniALBERT-128")
|
| 26 |
# For Sequence Classification use the below code
|
|
|
|
| 20 |
from minialbert_modeling import MiniAlbertForSequenceClassification, MiniAlbertForTokenClassification
|
| 21 |
```
|
| 22 |
Finally, load the model like a regular model in the transformers library using the below code:
|
| 23 |
+
```Python
|
| 24 |
# For NER use the below code
|
| 25 |
model = MiniAlbertForTokenClassification.from_pretrained("nlpie/miniALBERT-128")
|
| 26 |
# For Sequence Classification use the below code
|