Update README.md
Browse files
README.md
CHANGED
|
@@ -19,5 +19,5 @@ RoBERTa-small을 Codenet데이터로 pre_train한 가중치입니다.
|
|
| 19 |
## How to use
|
| 20 |
```python
|
| 21 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 22 |
-
tok = AutoTokenizer.from_pretrained("
|
| 23 |
-
mdl = AutoModelForSequenceClassification.from_pretrained("
|
|
|
|
| 19 |
## How to use
|
| 20 |
```python
|
| 21 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 22 |
+
tok = AutoTokenizer.from_pretrained("hosung1/roberta_small_mlm_from_scratch")
|
| 23 |
+
mdl = AutoModelForSequenceClassification.from_pretrained("hosung1/roberta_small_mlm_from_scratch")
|