Update src/ai_processor.py

src/ai_processor.py

@@ -212,9 +212,13 @@ def load_yolo_model():
     
 def load_segmentation_model(SEG_MODEL_PATH):
     """Lazy import and load segmentation model."""
-    from tensorflow.keras.models import load_model
+    # Use the lazy loader to ensure TF is imported correctly
+    load_model = _import_tf_loader()
+    # The compile=False argument is crucial for avoiding optimizer/loss issues
+    # and can sometimes help with layer deserialization.
     return load_model(SEG_MODEL_PATH, compile=False)
 
+
 def load_classification_pipeline():
     pipe = _import_hf_cls()
     return pipe("image-classification", model="Hemg/Wound-classification", token=HF_TOKEN, device="cpu")