Update src/ai_processor.py

src/ai_processor.py

@@ -211,10 +211,11 @@ def load_yolo_model():
     # Construct model with CUDA masked to avoid auto-selecting cuda:0
     with _no_cuda_env():
         model = YOLO(YOLO_MODEL_PATH)
-    return modeldef load_segmentation_model():
+    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})
+    return load_model(SEG_MODEL_PATH, compile=False, custom_objects={'InputLayer': tf.keras.layers.InputLayer})
 
 def load_classification_pipeline():
     pipe = _import_hf_cls()