updating item list

app.py

@@ -37,71 +37,60 @@ year = {2024}
 
 # List of fashion items
 items = [
-    "leggings", "jogger", "palazzo", "cargo", "dresspants", "chinos", 
-    "dress", "blouse", "t-shirt", "jeans", "skirt", "shorts", 
-    "sweater", "cardigan", "tank top", "hoodie", "coat", 
-    "jacket", "polo shirt", "crop top", "romper", 
-    "overalls", "blazer", "sweatpants", "vest", 
-    "dungarees", "poncho", "bodysuit", "maxi dress", 
-    "hat", "sunglasses", "glasses", "shoes", "sandals", "heels", "trainers", "belt", "tie", "dress shirt", "boots",
-    "slippers",
-    "sneakers",
-    "insoles",
-    "socks",
-    "insulated jacket",
-    "fleece",
-    "rain jacket",
-    "running jacket",
-    "windbreaker",
-    "shirt",
-    "t-shirt",
-    "tank top",
-    "graphic top",
-    "sweater",
-    "sweatshirt",
-    "vest",
-    "pant",
-    "legging",
-    "short",
-    "dress",
-    "skirt",
-    "skort",
-    "brief",
-    "sports bra",
-    "base layer top",
-    "base layer bottom",
-    "swimsuit",
-    "rashguard",
-    "water shorts",
-    "cover up",
-    "goggle",
-    "hat",
-    "sun hat",
-    "glove",
-    "mitten",
-    "belt",
-    "leg gaiter",
+    "Leggings", "Jogger", "Palazzo", "Cargo", "Dresspants", "Chinos", 
+    "Dress", "Blouse", "T-Shirt", "Jeans", "Skirt", "Shorts", 
+    "Sweater", "Cardigan", "Tank Top", "Hoodie", "Coat", 
+    "Jacket", "Polo Shirt", "Crop Top", "Romper", 
+    "Overalls", "Blazer", "Sweatpants", "Vest", 
+    "Dungarees", "Poncho", "Bodysuit", "Maxi Dress", 
+    "Hat", "Sunglasses", "Glasses", "Shoes", "Sandals", 
+    "Heels", "Trainers", "Belt", "Tie", "Dress Shirt", 
+    "Boots", "Slippers", "Sneakers", "Insoles", "Socks", 
+    "Insulated Jacket", "Fleece", "Rain Jacket", "Running Jacket", 
+    "Windbreaker", "Shirt", "Graphic Top", "Sweatshirt", 
+    "Pant", "Legging", "Short", "Skort", "Brief", 
+    "Sports Bra", "Base Layer Top", "Base Layer Bottom", 
+    "Swimsuit", "Rashguard", "Water Shorts", "Cover Up", 
+    "Goggle", "Sun Hat", "Glove", "Mitten", "Leg Gaiter", 
+    "Bandana", "Neck Gaiter", "Balaclava", "Sunglass", 
+    "Watch", "Bag", "Boxer", "Swim Trunk", "Ring", 
+    "Necklace", "Earing", "Pocket Watch", "Smock", 
+    "Trouser", "Tuxedo", "Cufflinks", "Suspenders", 
+    "Handkerchief", "Scarf", "Wristband", "Beanie", 
+    "Fedora", "Beret", "Clutch Bag", "Crossbody Bag", 
+    "Duffel Bag", "Backpack", "Umbrella", "Flip Flops", 
+    "Espadrilles", "Loafers", "Moccasins", "Chelsea Boots", 
+    "Ankle Boots", "Bow Tie", "Tie Clip", "Hair Clip", 
+    "Headband", "Visor", "Baseball Cap", "Bucket Hat", 
+    "Fingerless Gloves", "Touchscreen Gloves", "Trench Coat", 
+    "Peacoat", "Parka", "Bomber Jacket", "Utility Vest", 
+    "Puffer Jacket", "Cape", "Shrug", "Kimono", "Sarong", 
+    "Apron", "Bikini", "Poncho", "Wristwatch", "Choker", 
+    "Brooch", "Anklet", "Toe Ring", "Waist Belt"
 ]
 
+
 # Initialize the model and tokenizer
 model_name = 'hf-hub:Marqo/marqo-fashionSigLIP'
 model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(model_name)
 tokenizer = open_clip.get_tokenizer(model_name)
 
-# Generate descriptions dynamically
 def generate_description(item):
-    if "pants" in item or item in ["leggings", "jogger", "cargo", "chinos", "palazzo"]:
-        return f"a pair of {item} pants"
-    elif item in ["dress", "blouse", "t-shirt", "tank top", "sweater", "cardigan", "hoodie", "coat", "jacket", "polo shirt", "crop top", "romper", "blazer", "vest", "bodysuit", "maxi dress"]:
-        return f"a {item}"
-    elif item in ["hat", "sunglasses", "glasses"]:
-        return f"a {item} worn on the head or face"
-    elif item in ["shoes", "sandals", "heels", "trainers"]:
-        return f"a pair of {item} worn on the feet"
-    elif item in ["jeans", "skirt", "shorts", "sweatpants", "dungarees", "poncho", "overalls", "dress shirt"]:
-        return f"a {item} piece of clothing"
+    if "Pants" in item or item in ["Leggings", "Jogger", "Cargo", "Chinos", "Palazzo", "Dresspants", "Sweatpants", "Pant", "Legging", "Skort", "Trouser"]:
+        return f"A pair of {item} pants"
+    elif item in ["Dress", "Blouse", "T-Shirt", "Tank Top", "Sweater", "Cardigan", "Hoodie", "Coat", "Jacket", "Polo Shirt", "Crop Top", "Romper", "Blazer", "Vest", "Bodysuit", "Maxi Dress", "Graphic Top", "Shirt", "Base Layer Top", "Base Layer Bottom", "Swimsuit", "Rashguard", "Cover Up", "Smock", "Tuxedo"]:
+        return f"A {item}"
+    elif item in ["Hat", "Sunglasses", "Glasses", "Sun Hat", "Goggle", "Balaclava"]:
+        return f"A {item} worn on the head or face"
+    elif item in ["Shoes", "Sandals", "Heels", "Trainers", "Boots", "Slippers", "Sneakers", "Insoles", "Socks"]:
+        return f"A pair of {item} worn on the feet"
+    elif item in ["Jeans", "Skirt", "Shorts", "Dungarees", "Poncho", "Overalls", "Brief", "Boxer", "Swim Trunk", "Ring", "Necklace", "Earing", "Pocket Watch"]:
+        return f"A {item} piece of clothing"
+    elif item in ["Boxing Gloves", "Glove", "Mitten"]:
+        return f"An item of {item} worn on the hands"
     else:
-        return f"a fashion item called {item}"
+        return f"A fashion item called {item}"
+
 
 items_desc = [generate_description(item) for item in items]
 text = tokenizer(items_desc)