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 |
|