Added model and handler

handler.py

@@ -0,0 +1,108 @@
+import torch
+import torchvision.transforms as transforms
+from PIL import Image
+import json
+import os
+
+# Load Model
+MODEL_PATH = "model_scripted_efficientnet.pt"
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+model = torch.jit.load(MODEL_PATH, map_location=device)
+model.eval()
+
+# Define Transform
+transform = transforms.Compose([
+    transforms.Resize((224, 224)),
+    transforms.ToTensor(),
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
+])
+
+# Mappings
+issue_name_to_area_mapping = {
+    "Forehead Wrinkles": "Skin",
+    "Crow's Feet Wrinkles": "Skin",
+    "Glabella Wrinkles": "Skin",
+    "Under Eye Wrinkles": "Skin",
+    "Perioral Wrinkles": "Skin",
+    "Bunny Lines": "Skin",
+    "Neck Lines": "Skin",
+    "Dark Spots": "Skin",
+    "Red Spots": "Skin",
+    "Scars": "Skin",
+    "Dry Skin": "Skin",
+    "Whiteheads": "Skin",
+    "Blackheads": "Skin",
+    "Under Eye Dark Circles": "Skin",
+    "Crepey Skin": "Skin",
+    "Nasolabial Folds": "Skin",
+    "Marionette Lines": "Skin",
+    "Temporal Hollow": "Forehead",
+    "Brow Asymmetry": "Forehead",
+    "Flat Forehead": "Forehead",
+    "Brow Ptosis": "Forehead",
+    "Excess Upper Eyelid Skin": "Eyes",
+    "Lower Eyelid - Excess Skin": "Eyes",
+    "Upper Eyelid Droop": "Eyes",
+    "Lower Eyelid Sag": "Eyes",
+    "Lower Eyelid Bags": "Eyes",
+    "Under Eye Hollow": "Eyes",
+    "Upper Eye Hollow": "Eyes",
+    "Mid Cheek Flattening": "Cheeks",
+    "Cheekbone - Not Prominent": "Cheeks",
+    "Heavy Lateral Cheek": "Cheeks",
+    "Crooked Nose": "Nose",
+    "Droopy Tip": "Nose",
+    "Dorsal Hump": "Nose",
+    "Tip Droop When Smiling": "Nose",
+    "Thin Lips": "Lips",
+    "Lacking Philtral Column": "Lips",
+    "Long Philtral Column": "Lips",
+    "Gummy Smile": "Lips",
+    "Asymmetric Lips": "Lips",
+    "Dry Lips": "Lips",
+    "Lip Thinning When Smiling": "Lips",
+    "Retruded Chin": "Jawline",
+    "Over-Projected Chin": "Jawline",
+    "Asymmetric Chin": "Jawline",
+    "Jowls": "Jawline",
+    "Lower Cheeks - Volume Depletion": "Jawline",
+    "Ill-Defined Jawline": "Jawline",
+    "Asymmetric Jawline": "Jawline",
+    "Masseter Hypertrophy": "Jawline",
+    "Prejowl Sulcus": "Jawline",
+    "Loose Neck Skin": "Jawline",
+    "Platysmal Bands": "Jawline",
+    "Excess/Submental Fullness": "Jawline"
+}
+
+supported_issues = [
+    "Dark Spots",
+    "Dry Lips",
+    "Forehead Wrinkles",
+    "Jowls",
+    "Nasolabial Folds",
+    "Prejowl Sulcus",
+    "Thin Lips",
+    "Under Eye Hollow",
+    "Under Eye Wrinkles",
+    "Brow Asymmetry"
+]
+
+def predict(image_path):
+    image = Image.open(image_path).convert("RGB")
+    image = transform(image).unsqueeze(0).to(device)
+
+    with torch.no_grad():
+        outputs = model(image)
+    
+    predictions = outputs.squeeze().tolist()
+    output_labels = [label for label, prob in zip(supported_issues, predictions) if prob > 0.5]
+    
+    return {"predictions": output_labels}
+
+# Hugging Face API Inference Handler
+def handle_request(data):
+    image_path = data["image_path"]
+    results = predict(image_path)
+    return json.dumps(results)

model_scripted_efficientnet.pt

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

requirements.txt

@@ -0,0 +1,3 @@
+torch
+torchvision
+pillow