Update main.py

main.py

@@ -1,14 +1,5 @@
 import os
 import cv2
-# --- FIX FOR ULTRALYTICS HEADLESS OPENCV BUG ---
-# The headless version of OpenCV removes GUI features. Ultralytics forgets to check 
-# for this and tries to import cv2.imshow, causing a crash. This safely mocks it!
-if not hasattr(cv2, 'imshow'):
-    cv2.imshow = lambda *args, **kwargs: None
-    cv2.waitKey = lambda *args, **kwargs: None
-    cv2.destroyAllWindows = lambda *args, **kwargs: None
-    cv2.destroyWindow = lambda *args, **kwargs: None
-# -----------------------------------------------
 import numpy as np
 import base64
 import gc