Added Git LFS pull workaround

handler.py

@@ -1,13 +1,19 @@
+import os
+import subprocess
 import torch
 import torchvision.transforms as transforms
 from PIL import Image
 import json
-import os
 
-# Load Model
+# Fetch model from Git LFS if not downloaded
 MODEL_PATH = "model_scripted_efficientnet.pt"
-device = "cuda" if torch.cuda.is_available() else "cpu"
 
+if not os.path.exists(MODEL_PATH):
+    print("Model file not found! Fetching from Git LFS...")
+    subprocess.run(["git", "lfs", "pull"], check=True)
+
+# Load Model
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model = torch.jit.load(MODEL_PATH, map_location=device)
 model.eval()
 
@@ -18,64 +24,7 @@ transform = transforms.Compose([
     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 labels
 supported_issues = [
     "Dark Spots",
     "Dry Lips",
@@ -95,10 +44,10 @@ def predict(image_path):
 
     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