Update app.py
Browse files
app.py
CHANGED
@@ -23,27 +23,28 @@ LABEL_MAPPING = {
|
|
23 |
CATEGORY_NAMES = {
|
24 |
1: "Animals & Pet Supplies",
|
25 |
8: "Arts & Entertainment",
|
26 |
-
111: "
|
27 |
-
141: "
|
28 |
-
166: "
|
29 |
-
222: "
|
30 |
-
412: "
|
31 |
-
436: "
|
32 |
-
469: "
|
33 |
-
536: "
|
34 |
-
537: "
|
35 |
632: "Hardware",
|
36 |
-
772: "
|
37 |
-
783: "
|
38 |
-
888: "
|
39 |
-
922: "
|
40 |
988: "Sporting Goods",
|
41 |
-
1239: "
|
42 |
-
2092: "
|
43 |
-
5181: "
|
44 |
-
5605: "
|
45 |
}
|
46 |
|
|
|
47 |
# Reverse mapping for model output index to text label
|
48 |
INDEX_TO_CATEGORY = {v: f"[{k}] {CATEGORY_NAMES[k]}" for k, v in LABEL_MAPPING.items()}
|
49 |
|
|
|
23 |
CATEGORY_NAMES = {
|
24 |
1: "Animals & Pet Supplies",
|
25 |
8: "Arts & Entertainment",
|
26 |
+
111: "Business & Industrial",
|
27 |
+
141: "Cameras & Optics",
|
28 |
+
166: "Apparel & Accessories",
|
29 |
+
222: "Electronics",
|
30 |
+
412: "Food, Beverages & Tobacco",
|
31 |
+
436: "Furniture",
|
32 |
+
469: "Health & Beauty",
|
33 |
+
536: "Home & Garden",
|
34 |
+
537: "Baby & Toddler",
|
35 |
632: "Hardware",
|
36 |
+
772: "Mature",
|
37 |
+
783: "Media",
|
38 |
+
888: "Vehicles & Parts",
|
39 |
+
922: "Office Supplies",
|
40 |
988: "Sporting Goods",
|
41 |
+
1239: "Toys & Games",
|
42 |
+
2092: "Software",
|
43 |
+
5181: "Luggage & Bags",
|
44 |
+
5605: "Religious & Ceremonial"
|
45 |
}
|
46 |
|
47 |
+
|
48 |
# Reverse mapping for model output index to text label
|
49 |
INDEX_TO_CATEGORY = {v: f"[{k}] {CATEGORY_NAMES[k]}" for k, v in LABEL_MAPPING.items()}
|
50 |
|