Bugfix update model

app.py

@@ -12,9 +12,9 @@ from scipy.special import softmax
 logger = logging.getLogger("basebody")
 CLIP_MODEL_NAME = "ViT-B/16"
 
-TEXT_PROMPTS_FILE_NAME = "text_prompts.json"
+TEXT_PROMPTS_FILE_NAME = "text_prompts2.json"
 HAIR_TEXT_PROMPTS_FILE_NAME = "text_prompts_hair.json"
-LOGISTIC_REGRESSION_MODEL_FILE_NAME = "logistic_regression_l1_oct_2.pkl"
+LOGISTIC_REGRESSION_MODEL_FILE_NAME = "logistic_regression_l1_oct_26.pkl"
 HAIR_RF_CLASSIFIER_MODEL_FILE_NAME = "hairclassifier_rf.pkl"
 
 HF_TOKEN = os.getenv('HF_TOKEN')

logistic_regression_l1_oct_26.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d539f04fcb0566f83bbf1a9e06fbde0b0527bfba1ded77afe7855cb507e375c1
+size 1222

text_prompts2.json

@@ -0,0 +1,514 @@
+{
+  "clothing": [
+    "is character wearing clothing",
+    "is character not wearing clothing"
+  ],
+  "outfit": [
+    "is character wearing an outfit",
+    "is character not wearing an outfit"
+  ],
+  "equipment": [
+    "is character using equipment",
+    "is character not using equipment"
+  ],
+  "uniform": [
+    "is character wearing a uniform",
+    "is character not wearing a uniform"
+  ],
+  "jewelry": [
+    "is character wearing jewelry",
+    "is character not wearing jewelry"
+  ],
+  "jacket": [
+    "is character wearing a jacket",
+    "is character not wearing a jacket"
+  ],
+  "cardigan": [
+    "is character wearing a cardigan",
+    "is character not wearing a cardigan"
+  ],
+  "shoes": [
+    "is character wearing shoes",
+    "is character not wearing shoes"
+  ],
+  "jeans": [
+    "is character wearing jeans",
+    "is character not wearing jeans"
+  ],
+  "french": [
+    "is character wearing French-style clothing",
+    "is character not wearing French-style clothing"
+  ],
+  "chain": [
+    "is character wearing a chain",
+    "is character not wearing a chain"
+  ],
+  "weather": [
+    "is character wearing weather-appropriate clothing",
+    "is character not wearing weather-appropriate clothing"
+  ],
+  "trousers": [
+    "is character wearing trousers",
+    "is character not wearing trousers"
+  ],
+  "shorts": [
+    "is character wearing shorts",
+    "is character not wearing shorts"
+  ],
+  "tunic": [
+    "is character wearing a tunic",
+    "is character not wearing a tunic"
+  ],
+  "gown": [
+    "is character wearing a gown",
+    "is character not wearing a gown"
+  ],
+  "shirt": [
+    "is character wearing a shirt",
+    "is character not wearing a shirt"
+  ],
+  "hoodie": [
+    "is character wearing a hoodie",
+    "is character not wearing a hoodie"
+  ],
+  "crown": [
+    "is character wearing a crown",
+    "is character not wearing a crown"
+  ],
+  "suit": [
+    "is character wearing a suit",
+    "is character not wearing a suit"
+  ],
+  "armor": [
+    "is character wearing armor",
+    "is character not wearing armor"
+  ],
+  "kimono": [
+    "is character wearing a kimono",
+    "is character not wearing a kimono"
+  ],
+  "camping": [
+    "is character using camping equipment",
+    "is character not using camping equipment"
+  ],
+  "text": [
+    "is character wearing clothing with letters or numbers",
+    "is character not wearing clothing with letters or numbers"
+  ],
+  "blouse": [
+    "is character wearing a blouse",
+    "is character not wearing a blouse"
+  ],
+  "pajamas": [
+    "is character wearing pajamas",
+    "is character not wearing pajamas"
+  ],
+  "glasses": [
+    "is character wearing glasses",
+    "is character not wearing glasses"
+  ],
+  "cashmere": [
+    "is character wearing cashmere",
+    "is character not wearing cashmere"
+  ],
+  "knight": [
+    "is character dressed as a knight",
+    "is character not dressed as a knight"
+  ],
+  "fly": [
+    "is character wearing flying gear",
+    "is character not wearing flying gear"
+  ],
+  "overalls": [
+    "is character wearing overalls",
+    "is character not wearing overalls"
+  ],
+  "boxing": [
+    "is character wearing boxing gloves",
+    "is character not wearing boxing gloves"
+  ],
+  "lipstick": [
+    "is character wearing lipstick",
+    "is character not wearing lipstick"
+  ],
+  "pants": [
+    "is character wearing pants",
+    "is character not wearing pants"
+  ],
+  "apron": [
+    "is character wearing an apron",
+    "is character not wearing an apron"
+  ],
+  "zipper": [
+    "is character wearing something with a zipper",
+    "is character not wearing something with a zipper"
+  ],
+  "fighting": [
+    "is character wearing fighting gear",
+    "is character not wearing fighting gear"
+  ],
+  "headset": [
+    "is character wearing a headset",
+    "is character not wearing a headset"
+  ],
+  "lace": [
+    "is character wearing lace",
+    "is character not wearing lace"
+  ],
+  "bowling": [
+    "is character wearing bowling shoes",
+    "is character not wearing bowling shoes"
+  ],
+  "shawl": [
+    "is character wearing a shawl",
+    "is character not wearing a shawl"
+  ],
+  "flannel": [
+    "is character wearing flannel",
+    "is character not wearing flannel"
+  ],
+  "boots": [
+    "is character wearing boots",
+    "is character not wearing boots"
+  ],
+  "helmet": [
+    "is character wearing a helmet",
+    "is character not wearing a helmet"
+  ],
+  "t-shirt": [
+    "is character wearing a t-shirt",
+    "is character not wearing a t-shirt"
+  ],
+  "sandals": [
+    "is character wearing sandals",
+    "is character not wearing sandals"
+  ],
+  "jersey": [
+    "is character wearing a jersey",
+    "is character not wearing a jersey"
+  ],
+  "glove": [
+    "is character wearing gloves",
+    "is character not wearing gloves"
+  ],
+  "outdoors": [
+    "is character wearing outdoor gear",
+    "is character not wearing outdoor gear"
+  ],
+  "cargo": [
+    "is character wearing cargo pants",
+    "is character not wearing cargo pants"
+  ],
+  "tank top": [
+    "is character wearing a tank top",
+    "is character not wearing a tank top"
+  ],
+  "hair extensions": [
+    "is character wearing hair extensions",
+    "is character not wearing hair extensions"
+  ],
+  "scarf": [
+    "is character wearing a scarf",
+    "is character not wearing a scarf"
+  ],
+  "coat": [
+    "is character wearing a coat",
+    "is character not wearing a coat"
+  ],
+  "robe": [
+    "is character wearing a robe",
+    "is character not wearing a robe"
+  ],
+  "wizard": [
+    "is character dressed as a wizard",
+    "is character not dressed as a wizard"
+  ],
+  "backpack": [
+    "is character wearing a backpack",
+    "is character not wearing a backpack"
+  ],
+  "trenchcoat": [
+    "is character wearing a trenchcoat",
+    "is character not wearing a trenchcoat"
+  ],
+  "dress": [
+    "is character wearing a dress",
+    "is character not wearing a dress"
+  ],
+  "hat": [
+    "is character wearing a hat",
+    "is character not wearing a hat"
+  ],
+  "belt": [
+    "is character wearing a belt",
+    "is character not wearing a belt"
+  ],
+  "watch": [
+    "is character wearing a watch",
+    "is character not wearing a watch"
+  ],
+  "bag": [
+    "is character carrying a bag",
+    "is character not carrying a bag"
+  ],
+  "umbrella": [
+    "is character holding an umbrella",
+    "is character not holding an umbrella"
+  ],
+  "tattoo": [
+    "does character have a tattoo",
+    "does character not have a tattoo"
+  ],
+  "piercing": [
+    "does character have a piercing",
+    "does character not have a piercing"
+  ],
+  "bracelet": [
+    "is character wearing a bracelet",
+    "is character not wearing a bracelet"
+  ],
+  "necklace": [
+    "is character wearing a necklace",
+    "is character not wearing a necklace"
+  ],
+  "background-objects": [
+    "are there objects in the background",
+    "there are no objects in the background"
+  ],
+  "foreground-objects": [
+    "are there objects in the foreground",
+    "there are no objects in the foreground"
+  ],
+  "animals": [
+    "is there an animal near the character",
+    "there is no animal near the character"
+  ],
+  "vehicles": [
+    "is there a vehicle near the character",
+    "there is no vehicle near the character"
+  ],
+  "instruments": [
+    "is character holding a musical instrument",
+    "is character not holding a musical instrument"
+  ],
+  "weapons": [
+    "is character holding a weapon",
+    "is character not holding a weapon"
+  ],
+  "food-drink": [
+    "is character holding food or drink",
+    "is character not holding food or drink"
+  ],
+  "electronics": [
+    "is character using an electronic device",
+    "is character not using an electronic device"
+  ],
+  "books": [
+    "is character holding a book",
+    "is character not holding a book"
+  ],
+  "horns": [
+    "character has horns",
+    "character does not have horns"
+  ],
+  "skirt": [
+    "is character wearing a skirt",
+    "is character not wearing a skirt"
+  ],
+  "sweater": [
+    "is character wearing a sweater",
+    "is character not wearing a sweater"
+  ],
+  "sweatshirt": [
+    "is character wearing a sweatshirt",
+    "is character not wearing a sweatshirt"
+  ],
+  "sweatpants": [
+    "is character wearing sweatpants",
+    "is character not wearing sweatpants"
+  ],
+  "sunglasses": [
+    "is character wearing sunglasses",
+    "is character not wearing sunglasses"
+  ],
+  "tie": [
+    "is character wearing a tie",
+    "is character not wearing a tie"
+  ],
+  "skateboard": [
+    "is character using a skateboard",
+    "is character not using a skateboard"
+  ],
+  "skates": [
+    "is character using skates",
+    "is character not using skates"
+  ],
+  "rollerblades": [
+    "is character using rollerblades",
+    "is character not using rollerblades"
+  ],
+  "roller-skates": [
+    "is character using roller-skates",
+    "is character not using roller-skates"
+  ],
+  "surfboard": [
+    "is character using a surfboard",
+    "is character not using a surfboard"
+  ],
+  "snowboard": [
+    "is character using a snowboard",
+    "is character not using a snowboard"
+  ],
+  "skis": [
+    "is character using skis",
+    "is character not using skis"
+  ],
+  "goggles": [
+    "is character wearing goggles",
+    "is character not wearing goggles"
+  ],
+  "gloves": [
+    "is character wearing gloves",
+    "is character not wearing gloves"
+  ],
+  "mittens": [
+    "is character wearing mittens",
+    "is character not wearing mittens"
+  ],
+  "human": [
+    "is character human",
+    "is character not human"
+  ],
+  "animal": [
+    "is character an animal",
+    "is character not an animal"
+  ],
+  "hair": [
+    "does character have hair on its head",
+    "does character not have hair on its head"
+  ],
+  "bald": [
+    "is character bald",
+    "is character not bald"
+  ],
+  "headband": [
+    "is character wearing a headband",
+    "is character not wearing a headband"
+  ],
+  "tiara": [
+    "is character wearing a tiara",
+    "is character not wearing a tiara"
+  ],
+  "turban": [
+    "is character wearing a turban",
+    "is character not wearing a turban"
+  ],
+  "bandana": [
+    "is character wearing a bandana",
+    "is character not wearing a bandana"
+  ],
+  "hood": [
+    "is character wearing a hood",
+    "is character not wearing a hood"
+  ],
+  "wig": [
+    "is character wearing a wig",
+    "is character not wearing a wig"
+  ],
+  "headphones": [
+    "is character wearing headphones",
+    "is character not wearing headphones"
+  ],
+  "earmuffs": [
+    "is character wearing earmuffs",
+    "is character not wearing earmuffs"
+  ],
+  "veil": [
+    "is character wearing a veil",
+    "is character not wearing a veil"
+  ],
+  "feathers": [
+    "are there feathers on the character's head",
+    "there are no feathers on the character's head"
+  ],
+  "antenna": [
+    "does character have antenna on its head",
+    "does character not have antenna on its head"
+  ],
+  "head-decoration": [
+    "is there any decoration on the character's head",
+    "there is no decoration on the character's head"
+  ],
+  "skeleton": [
+    "is character a skeleton",
+    "is character not a skeleton"
+  ],
+  "robot": [
+    "is character a robot",
+    "is character not a robot"
+  ],
+  "cyborg": [
+    "is character a cyborg",
+    "is character not a cyborg"
+  ],
+  "mechanical-arm": [
+    "is character having a mechanical arm",
+    "is character not having a mechanical arm"
+  ],
+  "mechanical-leg": [
+    "is character having a mechanical leg",
+    "is character not having a mechanical leg"
+  ],
+  "exoskeleton": [
+    "is character wearing an exoskeleton",
+    "is character not wearing an exoskeleton"
+  ],
+  "bionic-eye": [
+    "does character have a bionic eye",
+    "does character not have a bionic eye"
+  ],
+  "mechanical-parts": [
+    "does character have visible mechanical parts",
+    "does character not have visible mechanical parts"
+  ],
+  "wires": [
+    "are there wires on or around the character",
+    "there are no wires on or around the character"
+  ],
+  "metallic-skin": [
+    "does character have metallic skin or surface",
+    "does character not have metallic skin or surface"
+  ],
+  "led-lights": [
+    "does character have LED lights or indicators",
+    "does character not have LED lights or indicators"
+  ],
+  "holographic-display": [
+    "does character have a holographic display or interface",
+    "does character not have a holographic display or interface"
+  ],
+  "battery-pack": [
+    "is character wearing or having a visible battery pack",
+    "is character not wearing or having a visible battery pack"
+  ],
+  "circuitry": [
+    "is visible circuitry present on the character",
+    "no visible circuitry is present on the character"
+  ],
+  "hydraulic-system": [
+    "does character have a visible hydraulic system",
+    "does character not have a visible hydraulic system"
+  ],
+  "cooling-fans": [
+    "does character have visible cooling fans or vents",
+    "does character not have visible cooling fans or vents"
+  ],
+  "control-panel": [
+    "does character have a control panel or interface",
+    "does character not have a control panel or interface"
+  ],
+  "gear-mechanism": [
+    "is there a visible gear mechanism on the character",
+    "there is no visible gear mechanism on the character"
+  ]
+}