diff --git a/modules/__init__.py b/modules/__init__.py
index 45a9cac..f6e8d86 100644
--- a/modules/__init__.py
+++ b/modules/__init__.py
@@ -11,8 +11,8 @@ def imwrite_unicode(path, img, params=None):
     root, ext = os.path.splitext(path)
     if not ext:
         ext = ".png"
-    result, encoded_img = cv2.imencode(ext, img, params if params else [])
     result, encoded_img = cv2.imencode(f".{ext}", img, params if params is not None else [])
+    if result:
         encoded_img.tofile(path)
         return True
     return False
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 6d9f8b8..81033cb 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,12 +10,20 @@ psutil==5.9.8
 tk==0.1.0
 customtkinter==5.2.2
 pillow==11.1.0
+pygrabber==0.1.4
+
+# Torch with CUDA for Windows/Linux and CPU-only fallback for macOS
 torch==2.5.1+cu118; sys_platform != 'darwin'
 torch==2.5.1; sys_platform == 'darwin'
+
 torchvision==0.20.1; sys_platform != 'darwin'
 torchvision==0.20.1; sys_platform == 'darwin'
+
+# Use only GPU-accelerated ONNX runtime
 onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
 onnxruntime-gpu==1.17; sys_platform != 'darwin'
+
 tensorflow; sys_platform != 'darwin'
+
 opennsfw2==0.10.2
 protobuf==4.23.2