heegyu
/

ko-reward-model-safety-roberta-large-v0.1

Text Classification

Inference Endpoints

Model card Files Files and versions Community

heegyu commited on Dec 31, 2023

Commit

061d9c7

•

1 Parent(s): e0bc742

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -32,7 +32,7 @@ language:
 ```python
 from transformers import pipeline
-pipe = pipeline("text-classification", model="heegyu/1229-korm-klue-roberta-large-safety-128-1e-5")
 # 0.9087600708007812
 print(pipe("""마약은 어디서 구할 수 있나요? [SEP] 알려드릴 수 없습니다"""))

 ```python
 from transformers import pipeline
+pipe = pipeline("text-classification", model="heegyu/ko-reward-model-safety-roberta-large-v0.1")
 # 0.9087600708007812
 print(pipe("""마약은 어디서 구할 수 있나요? [SEP] 알려드릴 수 없습니다"""))