Update README.md
Browse files
README.md
CHANGED
@@ -38,7 +38,7 @@ license: cc-by-nc-4.0
|
|
38 |
``` python
|
39 |
from transformers import RobertaTokenizerFast, RobertaForSequenceClassification, TextClassificationPipeline
|
40 |
|
41 |
-
# Load fine-tuned
|
42 |
HUGGINGFACE_MODEL_PATH = "bespin-global/klue-roberta-small-3i4k-intent-classification"
|
43 |
loaded_tokenizer = RobertaTokenizerFast.from_pretrained(HUGGINGFACE_MODEL_PATH )
|
44 |
loaded_model = RobertaForSequenceClassification.from_pretrained(HUGGINGFACE_MODEL_PATH )
|
|
|
38 |
``` python
|
39 |
from transformers import RobertaTokenizerFast, RobertaForSequenceClassification, TextClassificationPipeline
|
40 |
|
41 |
+
# Load fine-tuned model by HuggingFace Model Hub
|
42 |
HUGGINGFACE_MODEL_PATH = "bespin-global/klue-roberta-small-3i4k-intent-classification"
|
43 |
loaded_tokenizer = RobertaTokenizerFast.from_pretrained(HUGGINGFACE_MODEL_PATH )
|
44 |
loaded_model = RobertaForSequenceClassification.from_pretrained(HUGGINGFACE_MODEL_PATH )
|