Update bald_processor.py

bald_processor.py

@@ -2,6 +2,7 @@ import cv2
 import torch
 import numpy as np
 from PIL import Image
+import torch._dynamo
 from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
 from scipy.ndimage import label
 import logging
@@ -11,20 +12,14 @@ logger = logging.getLogger(__name__)
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 logger.info(f"Using device: {device}")
-
+torch._dynamo.config.suppress_errors = True
 try:
     logger.info("Loading SegFormer face-parsing model...")
     processor = SegformerImageProcessor.from_pretrained("jonathandinu/face-parsing")
     model = SegformerForSemanticSegmentation.from_pretrained("jonathandinu/face-parsing")
     model.to(device)
     model.eval()
-    
-    if torch.__version__.startswith('2'):
-        try:
-            model = torch.compile(model)
-            logger.info("Torch compile enabled → faster inference!")
-        except Exception as e:
-            logger.warning(f"Torch compile failed: {e}")
+
             
     logger.info("Model loaded successfully!")
 except Exception as e: