Update main.py

main.py

@@ -1,5 +1,14 @@
 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