dejanseo commited on
Commit
f60c09b
·
verified ·
1 Parent(s): f39e1d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -17
app.py CHANGED
@@ -23,27 +23,28 @@ LABEL_MAPPING = {
23
  CATEGORY_NAMES = {
24
  1: "Animals & Pet Supplies",
25
  8: "Arts & Entertainment",
26
- 111: "Apparel & Accessories",
27
- 141: "Baby & Toddler",
28
- 166: "Books & Magazines",
29
- 222: "Business & Industrial",
30
- 412: "Cameras & Optics",
31
- 436: "Cars & Vehicles",
32
- 469: "Computers & Electronics",
33
- 536: "Food & Beverages",
34
- 537: "Furniture",
35
  632: "Hardware",
36
- 772: "Health & Beauty",
37
- 783: "Home & Garden",
38
- 888: "Luggage & Bags",
39
- 922: "Media",
40
  988: "Sporting Goods",
41
- 1239: "Software",
42
- 2092: "Sports & Outdoors",
43
- 5181: "Toys & Games",
44
- 5605: "Travel & Tourism"
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