Update app.py

app.py

@@ -10,13 +10,14 @@ from concurrent.futures import ThreadPoolExecutor, as_completed, TimeoutError
 
 # -------------------- CONFIG --------------------
 model_paths = {
-    "pores": "pores.pt",
-    "pig1": "pig1.pt",       # ✅ added pigmentation model
+    "pores2": "pores2.pt",
+    "pig1": "pig1.pt",       # ✅ pigmentation model
     "combine": "combine.pt", # ✅ combine model
     "wrinkle": "wrinkle.pt",
 }
 default_conf_threshold = 0.05
-special_conf_threshold = 0.08   # ✅ used for combine & pig1
+special_conf_threshold = 0.08   # ✅ for combine & pig1
+pores2_conf_threshold = 0.12    # ✅ special for pores2
 imgsz = 1024
 
 # -------------------- INIT --------------------
@@ -79,8 +80,10 @@ def detect_skin_type_from_image(img):
 
 # -------------------- HELPERS --------------------
 def run_model(model_name, model, img, face_polygon, face_area):
-    # ✅ set special conf threshold for combine.pt & pig1.pt
-    if model_name in ["combine", "pig1"]:
+    # ✅ confidence logic
+    if model_name == "pores2":
+        conf = pores2_conf_threshold
+    elif model_name in ["combine", "pig1"]:
         conf = special_conf_threshold
     else:
         conf = default_conf_threshold