teo-sanchez commited on
Commit
6d6a227
β€’
1 Parent(s): 2543c35

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -2
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** | n/a |
34
- | **Author** | [n/a]() |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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