kfkas
/

RoBERTa-large-Detection-G2P

Text Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

kfkas commited on Jun 1, 2023

Commit

2f3cb15

•

1 Parent(s): 25c0033

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -22,7 +22,7 @@ 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(model_dir)
 model = RobertaForSequenceClassification.from_pretrained(model_dir).to(device)
 text = "월드커 파나은행 대표티메 행우늬 이달러 이영영장 선물"

 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)
 text = "월드커 파나은행 대표티메 행우늬 이달러 이영영장 선물"