Update src/ai_processor.py

src/ai_processor.py

@@ -394,10 +394,12 @@ def load_yolo_model():
     with _no_cuda_env():
         model = YOLO(YOLO_MODEL_PATH)
     return model
+    
 def load_segmentation_model():
-    import tensorflow as tf
-    load_model = _import_tf_loader()
-    return load_model(SEG_MODEL_PATH, compile=False, custom_objects={'InputLayer': tf.keras.layers.InputLayer})
+    import os; os.environ.setdefault("KERAS_BACKEND","tensorflow")
+    import tensorflow as tf; tf.config.set_visible_devices([], "GPU")
+    import keras
+    return keras.models.load_model("src/segmentation_model.keras", compile=False)
 
 def load_classification_pipeline():
     pipe = _import_hf_cls()