# Convert bytes to a PIL image if necessary

Image3DProcessor.py

@@ -46,6 +46,10 @@ class Image3DProcessor:
         # Create a new Rembg session
         rembg_session = rembg.new_session()
 
+        # Convert bytes to a PIL image if necessary
+        if isinstance(input_image, bytes):
+            input_image = Image.open(BytesIO(input_image))
+
         # Preprocess input image
         if preprocess_background:
             image = input_image.convert("RGB")
@@ -58,8 +62,9 @@ class Image3DProcessor:
                 image = set_white_background(image)
 
         image = resize_to_128(image)
+        
+    return image
 
-        return image
     
     @torch.no_grad()
     def reconstruct_and_export(self, image):