Update vtoonify_model.py

vtoonify_model.py

@@ -18,8 +18,7 @@ import gc
 import huggingface_hub
 import os
 import logging
-from PIL import Image
-
+from PIL import Image  # Importing Image from PIL
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -173,10 +172,10 @@ class Model():
         # Transform and crop the image
         transform_size = 256
         output_size = 256
-        img = PIL.Image.fromarray(image)
-        img = img.transform((transform_size, transform_size), PIL.Image.QUAD, (quad + 0.5).flatten(), PIL.Image.BILINEAR)
+        img = Image.fromarray(image)  # Corrected to use PIL.Image
+        img = img.transform((transform_size, transform_size), Image.QUAD, (quad + 0.5).flatten(), Image.BILINEAR)
         if output_size < transform_size:
-            img = img.resize((output_size, output_size), PIL.Image.ANTIALIAS)
+            img = img.resize((output_size, output_size), Image.ANTIALIAS)
 
         return np.array(img)
 
@@ -319,4 +318,4 @@ class Model():
         """Convert a tensor image to OpenCV format."""
         tmp = ((img.cpu().numpy().transpose(1, 2, 0) + 1.0) * 127.5).astype(np.uint8).copy()
         logging.debug(f"Converted image shape: {tmp.shape}, strides: {tmp.strides}")
-        return cv2.cvtColor(tmp, cv2.COLOR_RGB2BGR) 
+        return cv2.cvtColor(tmp, cv2.COLOR_RGB2BGR)