p1atdev commited on
Commit
e4465bd
·
verified ·
1 Parent(s): 624818f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -13,6 +13,29 @@ Converted from [SmilingWolf/wd-swinv2-tagger-v3](https://huggingface.co/SmilingW
13
 
14
  ## Example
15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
16
  ```py
17
  from PIL import Image
18
 
 
13
 
14
  ## Example
15
 
16
+ ### Pipeline
17
+
18
+ ```py
19
+ from transformers import pipeline
20
+
21
+ pipe = pipeline("image-classification", model="p1atdev/wd-swinv2-tagger-v3-hf", trust_remote_code=True)
22
+
23
+ print(pipe("sample.webp", top_k=20))
24
+ #[{'label': '1girl', 'score': 0.996832549571991},
25
+ # {'label': 'solo', 'score': 0.958362340927124},
26
+ # {'label': 'dress', 'score': 0.9418023228645325},
27
+ # {'label': 'hat', 'score': 0.9263725280761719},
28
+ # {'label': 'sitting', 'score': 0.9178062677383423},
29
+ # {'label': 'looking_up', 'score': 0.8978123068809509},
30
+ # {'label': 'short_hair', 'score': 0.8242536187171936},
31
+ # {'label': 'sky', 'score': 0.784591794013977},
32
+ # {'label': 'outdoors', 'score': 0.767611563205719},
33
+ # {'label': 'rating:general', 'score': 0.7561964988708496}]
34
+ ```
35
+
36
+ ### AutoModel
37
+
38
+
39
  ```py
40
  from PIL import Image
41