Update README.md
Browse files
README.md
CHANGED
|
@@ -28,9 +28,9 @@ FYI: You can replace the folder redirect with "erythropygia/bert-turkish-subject
|
|
| 28 |
import torch
|
| 29 |
from transformers import AutoModelForSequenceClassification AutoTokenizer, AutoConfig, pipeline
|
| 30 |
|
| 31 |
-
config_subject = AutoConfig.from_pretrained("
|
| 32 |
-
model_subject = AutoModelForSequenceClassification.from_pretrained('
|
| 33 |
-
tokenizer_subject = AutoTokenizer.from_pretrained("
|
| 34 |
|
| 35 |
device = torch.device('cpu') #or 'cuda'
|
| 36 |
|
|
|
|
| 28 |
import torch
|
| 29 |
from transformers import AutoModelForSequenceClassification AutoTokenizer, AutoConfig, pipeline
|
| 30 |
|
| 31 |
+
config_subject = AutoConfig.from_pretrained("erythropygia/bert-turkish-subject")
|
| 32 |
+
model_subject = AutoModelForSequenceClassification.from_pretrained('erythropygia/bert-turkish-subject', config=config_subject)
|
| 33 |
+
tokenizer_subject = AutoTokenizer.from_pretrained("erythropygia/bert-turkish-subject")
|
| 34 |
|
| 35 |
device = torch.device('cpu') #or 'cuda'
|
| 36 |
|