huolongguo10 commited on
Commit
7c300b1
1 Parent(s): 6b4fd97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -11,7 +11,7 @@ tags:
11
  ---
12
  # check_sec
13
  检查web参数安全性,支持多种payload(v0.1.2)
14
-
15
  ## 类型
16
  ```
17
  LABEL_0: secure
@@ -23,8 +23,8 @@ LABEL_1: insecure(可能包含payload)
23
  import transformers
24
  from transformers import BertTokenizer, DataCollatorWithPadding
25
  from transformers import AutoModelForSequenceClassification
26
- tokenizer = BertTokenizer.from_pretrained('huolongguo10/check_sec')
27
- model = AutoModelForSequenceClassification.from_pretrained('huolongguo10/check_sec', num_labels=2)
28
  import torch
29
  def check(text):
30
  inputs = tokenizer(text, return_tensors="pt")
 
11
  ---
12
  # check_sec
13
  检查web参数安全性,支持多种payload(v0.1.2)
14
+ 注意:该版本不再维护,请使用tiny版。
15
  ## 类型
16
  ```
17
  LABEL_0: secure
 
23
  import transformers
24
  from transformers import BertTokenizer, DataCollatorWithPadding
25
  from transformers import AutoModelForSequenceClassification
26
+ tokenizer = BertTokenizer.from_pretrained('huolongguo10/check_sec_tiny')
27
+ model = AutoModelForSequenceClassification.from_pretrained('huolongguo10/check_sec_tiny', num_labels=2)
28
  import torch
29
  def check(text):
30
  inputs = tokenizer(text, return_tensors="pt")