RJuro commited on
Commit
042b1f4
1 Parent(s): dd424f7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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/danish-bert-go-aemotion')
61
- tokenizer = AutoTokenizer.from_pretrained('RJuro/danish-bert-go-aemotion')
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/danish-bert-go-aemotion')
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
  ```