Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
|
43 |
import torch
|
44 |
|
45 |
# Load model and tokenizer
|
46 |
-
model_name = "
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
48 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
49 |
|
|
|
43 |
import torch
|
44 |
|
45 |
# Load model and tokenizer
|
46 |
+
model_name = "SIRIS-Lab/org_type_classifier"
|
47 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
48 |
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
49 |
|