huolongguo10
/

check_sec_tiny

Text Classification

Model card Files Files and versions

huolongguo10 commited on Jul 17, 2023

Commit

39be4fe

·

1 Parent(s): 064236a

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -23,8 +23,8 @@ LABEL_1: insecure(可能包含payload)
 import transformers
 from transformers import BertTokenizer, DataCollatorWithPadding
 from transformers import AutoModelForSequenceClassification
-tokenizer = BertTokenizer.from_pretrained('huolongguo10/check_sec')
-model = AutoModelForSequenceClassification.from_pretrained('huolongguo10/check_sec', num_labels=2)
 import torch
 def check(text):
     inputs = tokenizer(text, return_tensors="pt")

 import transformers
 from transformers import BertTokenizer, DataCollatorWithPadding
 from transformers import AutoModelForSequenceClassification
+tokenizer = BertTokenizer.from_pretrained('huolongguo10/check_sec_tiny')
+model = AutoModelForSequenceClassification.from_pretrained('huolongguo10/check_sec_tiny', num_labels=2)
 import torch
 def check(text):
     inputs = tokenizer(text, return_tensors="pt")