Update binary_segmentation.py

binary_segmentation.py

@@ -14,6 +14,7 @@ import torch.nn.functional as F
 from PIL import Image
 from torchvision import transforms
 import cv2
+import torch.quantization
 
 # Configure logging
 logging.basicConfig(
@@ -530,7 +531,7 @@ class BinarySegmenter:
         
             # ✅ QUANTIZE to INT8 for CPU speedup
             if DEVICE == "cpu":
-                import torch.quantization
+                
                 self.model = torch.quantization.quantize_dynamic(
                     self.model,
                     {torch.nn.Linear, torch.nn.Conv2d},