Update README.md
Browse files
README.md
CHANGED
@@ -45,8 +45,8 @@ To use the model:
|
|
45 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
46 |
from transformers import pipeline
|
47 |
|
48 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
49 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
50 |
|
51 |
comp_classifier = pipeline('text-classification', model=model, tokenizer=tokenizer)
|
52 |
```
|
|
|
45 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
46 |
from transformers import pipeline
|
47 |
|
48 |
+
model = AutoModelForSequenceClassification.from_pretrained("nestauk/jobbert-base-cased-compdecs")
|
49 |
+
tokenizer = AutoTokenizer.from_pretrained("nestauk/jobbert-base-cased-compdecs")
|
50 |
|
51 |
comp_classifier = pipeline('text-classification', model=model, tokenizer=tokenizer)
|
52 |
```
|