Spaces:
Runtime error
Runtime error
Marcus Posey
commited on
Commit
·
806d130
1
Parent(s):
9fa792a
Update categories
Browse files
app.py
CHANGED
@@ -19,10 +19,10 @@ BOOK_MAPPING = {
|
|
19 |
}
|
20 |
|
21 |
CATEGORY_MAPPING = {
|
22 |
-
"
|
|
|
23 |
"Relationship": "relationship",
|
24 |
-
"
|
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"]
|