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

25c0033

•

1 Parent(s): 2f1f377

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,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)
 text = "월드커 파나은행 대표티메 행우늬 이달러 이영영장 선물"
 with torch.no_grad():

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