Commit
•
cf3881c
1
Parent(s):
7e06d8c
Update README.md
Browse files
README.md
CHANGED
|
@@ -21,9 +21,7 @@ You can use the model with a pipeline for text classification:
|
|
| 21 |
|
| 22 |
```python
|
| 23 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
| 24 |
-
from transformers.pipelines.pt_utils import KeyDataset
|
| 25 |
import datasets
|
| 26 |
-
from tqdm.auto import tqdm
|
| 27 |
|
| 28 |
tokenizer_name = "ESGBERT/SocRoBERTa-social"
|
| 29 |
model_name = "ESGBERT/SocRoBERTa-social"
|
|
|
|
| 21 |
|
| 22 |
```python
|
| 23 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
|
|
|
| 24 |
import datasets
|
|
|
|
| 25 |
|
| 26 |
tokenizer_name = "ESGBERT/SocRoBERTa-social"
|
| 27 |
model_name = "ESGBERT/SocRoBERTa-social"
|