Update README.md
Browse files
README.md
CHANGED
@@ -12,7 +12,7 @@ from transformers import XLMRobertaForSequenceClassification
|
|
12 |
|
13 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
14 |
|
15 |
-
model = XLMRobertaForSequenceClassification.from_pretrained("
|
16 |
model.eval()
|
17 |
model.to(device)
|
18 |
|
@@ -38,7 +38,7 @@ from transformers import XLMRobertaForSequenceClassification
|
|
38 |
|
39 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
40 |
|
41 |
-
model = AutoModel.from_pretrained("
|
42 |
model.eval()
|
43 |
model.to(device)
|
44 |
|
|
|
12 |
|
13 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
14 |
|
15 |
+
model = XLMRobertaForSequenceClassification.from_pretrained("LDKSolutions/KR-cryptoroberta-base", num_labels=3)
|
16 |
model.eval()
|
17 |
model.to(device)
|
18 |
|
|
|
38 |
|
39 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
40 |
|
41 |
+
model = AutoModel.from_pretrained("LDKSolutions/KR-cryptoroberta-base")
|
42 |
model.eval()
|
43 |
model.to(device)
|
44 |
|