kfkas
/

RoBERTa-large-Detection-G2P

Text Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

kfkas commited on Jul 26, 2023

Commit

2b05e1c

•

1 Parent(s): 3ffc0e2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -21,7 +21,7 @@ import torch
 import numpy as np
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_dir = "kfkas/RoBERTa-large-Detection-P2G"
 tokenizer = AutoTokenizer.from_pretrained('klue/roberta-large')
 model = RobertaForSequenceClassification.from_pretrained(model_dir).to(device)

 import numpy as np
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_dir = "kfkas/RoBERTa-large-Detection-G2P"
 tokenizer = AutoTokenizer.from_pretrained('klue/roberta-large')
 model = RobertaForSequenceClassification.from_pretrained(model_dir).to(device)