Update README.md
Browse files
README.md
CHANGED
|
@@ -57,8 +57,8 @@ Easiest use with `transformers` and `pipeline`:
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
| 59 |
|
| 60 |
-
model = AutoModelForSequenceClassification.from_pretrained('RJuro/
|
| 61 |
-
tokenizer = AutoTokenizer.from_pretrained('RJuro/
|
| 62 |
|
| 63 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 64 |
|
|
@@ -72,7 +72,7 @@ classifier('jeg elsker dig')
|
|
| 72 |
```python
|
| 73 |
from simpletransformers.classification import MultiLabelClassificationModel
|
| 74 |
|
| 75 |
-
model = MultiLabelClassificationModel('bert', 'RJuro/
|
| 76 |
|
| 77 |
predictions, raw_outputs = model.predict(df['text'])
|
| 78 |
```
|
|
|
|
| 57 |
```python
|
| 58 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
|
| 59 |
|
| 60 |
+
model = AutoModelForSequenceClassification.from_pretrained('RJuro/Da-HyggeBERT')
|
| 61 |
+
tokenizer = AutoTokenizer.from_pretrained('RJuro/Da-HyggeBERT')
|
| 62 |
|
| 63 |
classifier = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
|
| 64 |
|
|
|
|
| 72 |
```python
|
| 73 |
from simpletransformers.classification import MultiLabelClassificationModel
|
| 74 |
|
| 75 |
+
model = MultiLabelClassificationModel('bert', 'RJuro/Da-HyggeBERT')
|
| 76 |
|
| 77 |
predictions, raw_outputs = model.predict(df['text'])
|
| 78 |
```
|