Update README.md
Browse files
README.md
CHANGED
@@ -24,17 +24,22 @@ pipe = pipeline(
|
|
24 |
trust_remote_code=True,
|
25 |
)
|
26 |
|
27 |
-
print(pipe("sample.webp", top_k=
|
28 |
-
#[{'label': '1girl', 'score': 0.
|
29 |
-
# {'label': 'solo', 'score': 0.
|
30 |
-
# {'label': 'dress', 'score': 0.
|
31 |
-
# {'label': 'hat', 'score': 0.
|
32 |
-
# {'label': '
|
33 |
-
# {'label': '
|
34 |
-
# {'label': '
|
35 |
-
# {'label': '
|
36 |
-
# {'label': '
|
37 |
-
# {'label': '
|
|
|
|
|
|
|
|
|
|
|
38 |
```
|
39 |
|
40 |
### AutoModel
|
|
|
24 |
trust_remote_code=True,
|
25 |
)
|
26 |
|
27 |
+
print(pipe("sample.webp", top_k=15))
|
28 |
+
#[{'label': '1girl', 'score': 0.9973934888839722},
|
29 |
+
# {'label': 'solo', 'score': 0.9719744324684143},
|
30 |
+
# {'label': 'dress', 'score': 0.9539461135864258},
|
31 |
+
# {'label': 'hat', 'score': 0.9511678218841553},
|
32 |
+
# {'label': 'outdoors', 'score': 0.9438753128051758},
|
33 |
+
# {'label': 'sky', 'score': 0.9195725917816162},
|
34 |
+
# {'label': 'sitting', 'score': 0.9178725481033325},
|
35 |
+
# {'label': 'looking up', 'score': 0.9122412800788879},
|
36 |
+
# {'label': 'short hair', 'score': 0.8630313873291016},
|
37 |
+
# {'label': 'cloud', 'score': 0.8609118461608887},
|
38 |
+
# {'label': 'brown hair', 'score': 0.7723952531814575},
|
39 |
+
# {'label': 'short sleeves', 'score': 0.7649227380752563},
|
40 |
+
# {'label': 'day', 'score': 0.7641971111297607},
|
41 |
+
# {'label': 'rating:general', 'score': 0.7605368494987488},
|
42 |
+
# {'label': 'white dress', 'score': 0.7596388459205627}]
|
43 |
```
|
44 |
|
45 |
### AutoModel
|