teo-sanchez
commited on
Commit
β’
6d6a227
1
Parent(s):
2543c35
Update README.md
Browse files
README.md
CHANGED
@@ -30,8 +30,36 @@ model-index:
|
|
30 |
| **Components** | `tok2vec`, `ner` |
|
31 |
| **Vectors** | 514157 keys, 514157 unique vectors (300 dimensions) |
|
32 |
| **Sources** | n/a |
|
33 |
-
| **License** |
|
34 |
-
| **Author** | [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
### Label Scheme
|
37 |
|
|
|
30 |
| **Components** | `tok2vec`, `ner` |
|
31 |
| **Vectors** | 514157 keys, 514157 unique vectors (300 dimensions) |
|
32 |
| **Sources** | n/a |
|
33 |
+
| **License** | MIT |
|
34 |
+
| **Author** | [Selas.ai](www.selas.ai) |
|
35 |
+
|
36 |
+
### Description
|
37 |
+
|
38 |
+
Name entity recognition model to analyzing text-to-image prompts (Stable Diffusion).
|
39 |
+
|
40 |
+
The entities comprise 7 main categories and 11 subcategories for a total of 16 categories, extracted from a topic analysis made with [BERTopic](https://maartengr.github.io/BERTopic/index.html).
|
41 |
+
The topic analysis can be explored [the following visualization](https://teo-sanchez.github.io/projects/prompting_map.html).
|
42 |
+
|
43 |
+
βββ medium/
|
44 |
+
β βββ photography
|
45 |
+
β βββ painting
|
46 |
+
β βββ rendering
|
47 |
+
β βββ illustration
|
48 |
+
βββ influence/
|
49 |
+
β βββ artist
|
50 |
+
β βββ genre
|
51 |
+
β βββ artwork
|
52 |
+
β βββ repository
|
53 |
+
βββ light
|
54 |
+
βββ color
|
55 |
+
βββ composition
|
56 |
+
βββ detail
|
57 |
+
βββ context/
|
58 |
+
βββ era
|
59 |
+
βββ weather
|
60 |
+
βββ emotion
|
61 |
+
|
62 |
+
Prompt data are from the [diffusionDB](https://poloclub.github.io/diffusiondb/) database and were annotated by hand using [Prodigy](https://prodi.gy/).
|
63 |
|
64 |
### Label Scheme
|
65 |
|