pogzyb commited on
Commit
20b76b3
1 Parent(s): 589d6c4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
README.md CHANGED
@@ -1,3 +1,30 @@
1
  ---
2
  license: cc-by-4.0
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-4.0
3
  ---
4
+ # MarkupLM-Phish
5
+
6
+ The [MarkupLM]('https://huggingface.co/docs/transformers/v4.39.1/model_doc/markuplm') base model fine-tuned for phish classification!
7
+
8
+ ## Example Pipeline
9
+ ```python
10
+ import requests
11
+ from transformers import AutoProcessor, pipeline
12
+
13
+ phish_processor = AutoProcessor.from_pretrained(
14
+ "pogzyb/markuplm-phish", trust_remote_code=True
15
+ )
16
+ phish_pipeline = pipeline(
17
+ "text-classification",
18
+ model="pogzyb/markuplm-phish",
19
+ tokenizer=phish_processor,
20
+ max_length=512,
21
+ padding="max_length",
22
+ truncation=True,
23
+ )
24
+ html_string = requests.get("https://huggingface.co").text
25
+ phish_pipeline(html_string)
26
+ ```
27
+
28
+ ```
29
+ [{'label': 'negative', 'score': 0.9991624355316162}]
30
+ ```