Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ We introduce the model for multilabel ESG risks classification. There is 47 clas
|
|
| 9 |
## Usage
|
| 10 |
```python
|
| 11 |
|
| 12 |
-
from transformers import MPNetPreTrainedModel, MPNetModel
|
| 13 |
import torch
|
| 14 |
#Mean Pooling - Take attention mask into account for correct averaging
|
| 15 |
def mean_pooling(model_output, attention_mask):
|
|
|
|
| 9 |
## Usage
|
| 10 |
```python
|
| 11 |
|
| 12 |
+
from transformers import MPNetPreTrainedModel, MPNetModel, AutoTokenizer
|
| 13 |
import torch
|
| 14 |
#Mean Pooling - Take attention mask into account for correct averaging
|
| 15 |
def mean_pooling(model_output, attention_mask):
|