LDKSolutions
/

KR-cryptoroberta-base

Text Classification

Model card Files Files and versions

Lukekim commited on Sep 4, 2023

Commit

8a92d2b

·

1 Parent(s): 306a68c

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -12,7 +12,7 @@ from transformers import XLMRobertaForSequenceClassification
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-model = XLMRobertaForSequenceClassification.from_pretrained("axiomlabs/KR-cryptoroberta-base", num_labels=3)
 model.eval()
 model.to(device)
@@ -38,7 +38,7 @@ from transformers import XLMRobertaForSequenceClassification
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-model = AutoModel.from_pretrained("axiomlabs/KR-cryptoroberta-base")
 model.eval()
 model.to(device)

 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+model = XLMRobertaForSequenceClassification.from_pretrained("LDKSolutions/KR-cryptoroberta-base", num_labels=3)
 model.eval()
 model.to(device)
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+model = AutoModel.from_pretrained("LDKSolutions/KR-cryptoroberta-base")
 model.eval()
 model.to(device)