Marcus Posey commited on
Commit
806d130
·
1 Parent(s): 9fa792a

Update categories

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,10 +19,10 @@ BOOK_MAPPING = {
19
  }
20
 
21
  CATEGORY_MAPPING = {
22
- "Character": "character",
 
23
  "Relationship": "relationship",
24
- "Plot": "plot",
25
- "Numerical": "numerical"
26
  }
27
 
28
  MODEL_VARIANTS = ["rephrase", "rephrase_summarize", "entigraph"]
 
19
  }
20
 
21
  CATEGORY_MAPPING = {
22
+ "Identity": "identity",
23
+ "Motivation": "motivation",
24
  "Relationship": "relationship",
25
+ "Event": "event",
 
26
  }
27
 
28
  MODEL_VARIANTS = ["rephrase", "rephrase_summarize", "entigraph"]